apex库安装，解决amp not installed 报错( ModuleNotFoundError: No module named ‘apex‘)_modulenotfounderror: no module named 'apex

问题描述：

这是缺少apex库，需要安装！注意：不要使用pip install apex，虽然也能成功安装，但是最后运行程序会发现依旧出错，用不了

具体安装步骤如下：

# 执行git下载apex文件夹时，如果下载太慢，可以手动进入网址https://github.com/NVIDIA/apex下下载解压后执行后续操作
git clone https://github.com/NVIDIA/apex
cd apex
python setup.py install
1
2
3
4

执行后如果出现下图所示结果，表示安装成功！！

• 补充

# error
Traceback (most recent call last):
File "setup.py", line 35, in <module>
_, bare_metal_major, _ = get_cuda_bare_metal_version(CUDA_HOME)
File "setup.py", line 14, in get_cuda_bare_metal_version
raw_output = subprocess.check_output([cuda_dir + "/bin/nvcc", "-V"], universal_newlines=True)
TypeError: unsupported operand type(s) for +: 'NoneType' and 'str'
1
2
3
4
5
6
7

是因为conda安装的cuda是不全的，没有nvcc，解决方案conda install -c nvidia cuda-nvcc，安装成功！！

记得收藏哟，防止下次找不到啦！
看在我这么认真的份上，点个赞再走吧