pointnet2.pytorch(pointnet++)运行 python setup.py install 相关报错解决办法:

cuda11.0 cadnn8.0.5 pytorch=1.7.1 python=3.6

pointnet2.pytorch (sshaoshuai) 运行 python setup.py install 报错解决办法:

1、所有cpp/h文件中“AT_CHECK”修改为“TORCH_CHECK
2、所有cpp/h文件中“type().is_cuda()”修改为“is_cuda()
3、所有cpp/h文件中“.data()”修改为“.data_ptr()”
4、所有cpp/h文件中修改为“cudaStream_t stream = c10::cuda::getCurrentCUDAStream();”
5、computer 86问题解决:
先运行如下命令:export TORCH_CUDA_ARCH_LIST="8.0"

再执行:python setup.py install

你可能感兴趣的:(pytorch,python,深度学习)