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<float/int>()”修改为“.data_ptr<float/int>()”
4、所有cpp/h文件中修改为“cudaStream_t stream = c10::cuda::getCurrentCUDAStream();”
5、computer 86问题解决:
先运行如下命令:export TORCH_CUDA_ARCH_LIST="8.0"
再执行:python setup.py install