使用了这样的语句:
from apex.fp16_utils import *
from apex import amp, optimizers
发现没有这个依赖就使用了:
pip install apex
安装好后.fp16_util依然报错,但是打开pycharm的解释器你会发现这个apex不是github上nvidia的apex,只是重名了。
可以看这个链接:
链接: 引用apex时:ImportError: cannot import name ‘UnencryptedCookieSessionFactoryConfig’ from ‘pyramid.session’.
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir ./