编译chamfer3D报错,编译成功后找不到chamfer3D

一、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

问题解决

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值