nvcc fatal : Unsupported gpu architecture ‘compute_37‘

一、问题

        在复现Point_mae时,安装pointnet2_ops_lib模块一直安装不上,报错

        nvcc fatal : Unsupported gpu architecture 'compute_37'

        subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1.
        等等一长串错误

二、解决

1、最开始解决ninja -v这个错误,后来一直没成功。

2、之后看看nvcc这个错误,说是不支持算力,最后打开pointnet2_ops_lib中的setup.py文件,

将这个3.7+PTX删了,在安装就好了,误打误撞,折磨我好久。

我觉得可能是这个3.7太低了。

最后附上我的环境:

 Ubuntu系统+4090+python3.7+pytorch1.8.0+cuda10.2

评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值