要说明:在论文代码复现的过程中,环境要求是要安装torch的版本是1.4.0 但是当安装好之后,会报错,提示你torch版本过低,要不然安装1.6以上,要不然安装 NVIDIA apex
想都不用想,当然安装apex(害怕安装了高版本的torch会有其他错误,尽量避免)
apex安装方法:
git clone https://github.com/NVIDIA/apex
cd apex
python3 setup.py install
安装好之后运行代码,报错:
ModuleNotFoundError: No module named ‘amp_C‘
链接找到 utils.py 文件
方法 : 将import_amp_c
注释掉。
如下图
保存,然后代码就可以运行了。
解决办法,