cuda11.0 cadnn8.0.5 pytorch=1.7.1 python=3.6
pointnet2.pytorch编译 python setup.py install 报错解决办法:
1、所有cpp/h文件中“AT_CHECK”修改为“TORCH_CHECK”
2、所有cpp/h文件中“type().is_cuda()”修改为“is_cuda()”
3、所有cpp/h文件中“.data
4、所有cpp/h文件中修改为“cudaStream_t stream = c10::cuda::getCurrentCUDAStream();”
5、computer 86问题解决:
先运行如下命令:export TORCH_CUDA_ARCH_LIST="8.0" (此处改成7.5不行,必须改成8.0)
再执行:python setup.py install



