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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值