Ubuntu安装apex

#安装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