【环境配置】安装apex遇到的问题

环境配置】安装apex遇到的问题"/>

【环境配置】安装apex遇到的问题

问题

Traceback (most recent call last):File "<string>", line 1, in <module>File "/tmp/pip-req-build-md7yngpi/setup.py", line 35, in <module>_, bare_metal_major, _ = get_cuda_bare_metal_version(cpp_extension.CUDA_HOME)File "/tmp/pip-req-build-md7yngpi/setup.py", line 14, in get_cuda_bare_metal_versionraw_output = subprocess.check_output([cuda_dir + "/bin/nvcc", "-V"], universal_newlines=True)
TypeError: unsupported operand type(s) for +: 'NoneType' and 'str'

解决方法

在python setup.py install之前运行下面的命令

export TORCH_CUDA_ARCH_LIST="compute capability"