一、python setup.py install编译chamfer3D报错
出现nvcc fatal : Unsupported gpu architecture ‘compute_86‘的问题,是因为显卡与cuda版本支持的算力不匹配。
nvcc fatal : Unsupported gpu architecture ‘compute_86’ ninja: build
stopped: subcommand failed.
向系统环境变量中添加一项:TORCH_CUDA_ARCH_LIST,值为7.5
export TORCH_CUDA_ARCH_LIST="7.5"
问题解决
2、import chamfer3D 出现 “ModuleNotFoundError: No module named ‘chamfer3D’”
import sys
sys.path.append("submodules/ChamferDistancePytorch")
import torch
import chamfer3D.dist_chamfer_3D
问题解决