ERROR: Failed building wheel for pointnet2_ops
/CUDA kernel failed : no kernel image is available for execution on the device
是同一个,巨大坑
网上说法把3.7+PTX去掉,的确编译成功,但没有解决根本,后面可能cuda无法使用
出现错误CUDA kernel failed : no kernel image is available for execution on the device
网上说法Comment out all "CUDA_CHECK_ERRORS();" in .cu files in "_ext-src/src/" in PointNet2 project.
最终没有用
解决方法:
此处cuda 环境变量 TORCH_CUDA_ARCH_LIST需要根据自身环境变量设置,因cuda更新,最新的没有写上去,调用命令查看 我是cuda12.1,torch2.2
python -c "import torch; print(torch.cuda.get_device_capability())"
输出8,9 则添加8.9到TORCH_CUDA_ARCH_LIST,会报错的去掉3.7+PTX
最终编译成功,cuda也可以使用