原文链接
GPU:3060
CUDA11.4
torch1.7
原文是安装apex的问题,我是运行setup.py遇到的问题,原因都一样,3090的算力能达到8.6,但是CUDA11.4支持不了这么高的算力
在运行setup.py前,在终端输入
export TORCH_CUDA_ARCH_LIST="7.5"
再正常运行setup.py即可
这里的7.5指的是算力,是2080的。算力在这里看:https://developer.nvidia.com/zh-cn/cuda-gpus#compute
------------------------更新---------------------------
3090Ti不用加这句 直接setup.py就行
3090Ti配置:torch1.10+cu11.3