nvcc fatal : Unsupported gpu architecture ‘compute_86‘解决方法

原文链接
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

你可能感兴趣的:(杂谈,3060,CUDA,nvcc)