一、问题
在复现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