执行 echo $TORCH_CUDA_ARCH_LIST,发现CUDA架构版本包括
5.2 6.0 6.1 7.0 7.5 8.0 8.6 9.0+PTX
报错显示不支持(9.0+PTX),设置环境变量去掉即可:
TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6+PTX" pip install xxxx。 成功安装
执行 echo $TORCH_CUDA_ARCH_LIST,发现CUDA架构版本包括
5.2 6.0 6.1 7.0 7.5 8.0 8.6 9.0+PTX
报错显示不支持(9.0+PTX),设置环境变量去掉即可:
TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6+PTX" pip install xxxx。 成功安装