#安装apex混合精度工具
git clone https://github.com/NVIDIA/apex
cd apex
python setup.py install
出现报错:
torch.__version__ = 1.8.1+cu101
Traceback (most recent call last):
File "setup.py", line 136, in <module>
_, bare_metal_version = get_cuda_bare_metal_version(CUDA_HOME)
File "setup.py", line 23, 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'
可能是nvcc还没有安装
conda install -c nvidia cuda-nvcc
再执行安装步骤:
python setup.py install
即可成功安装
.....
Finished processing dependencies for apex==0.1