pointMLP安装point_ops_lib/.模块(pytorch版本)

pointMLP在github源码链接:ma-xu/pointMLP-pytorch: [ICLR 2022 poster] Official PyTorch implementation of "Rethinking Network Design and Local Geometry in Point Cloud: A Simple Residual MLP Framework" (github.com)

其中pointnet2_ops_lib需要单独配置。Point Cloud Transformer, SnowflakeNet, PointTr等点云深度学习相关的源码中都有出现。

在满足源码Requirements的环境后,在pytorch环境中CD到 pointnet2_ops_lib所在目录,然后运行 pip install pointnet2_ops_lib/.可能会出现报错。
 

基本安装一共分两步:

preparing metadata (setup.py)... 和building wheels for collected packages:pointnet2-ops

第一步出错的我那时候是重新安装了NVIDIA CUDA之后正常电脑环境变量里会自动出现CUDA_PATH和对应的cuda的版本

第二部=步我出错的问题在于RuntimeError:The detected CUDA version mismatches the version that was used to compile PyTorch.如果出现这种问题只需要到pytorch官网安装对应版本的pytorch即可。看自己cuda版本可以进入电脑环境变量里看CUDA_PATH,之前看教程还需要添加环境变量CUDA_HOME,之后没配置也成功了,如果没成功可以试一下是不是这个问题

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值