REGTR: End-to-end Point Cloud Correspondences with Transformers环境配置

最近看到这篇CVPR2022的文章,想跑一下实验看看,但是环境配置上踩了大坑

项目网址:GitHub - yewzijian/RegTR: End-to-end Point Cloud Correspondences with Transformersz

作者的要求如下:

 接下来进行环境配置

创建conda环境

注意一定要根据作者的python版本来,不然跑不了

conda create -n 环境名 python=3.8.8

conda下安装cuda11.1,这一步就不赘述了,网上一搜很多

pytorch安装

我先安装了他说的Pytorch1.9.1版本,然后一系列操作下来发现出现各种版本冲突问题,查阅博客http://t.csdn.cn/hGAtz后试了一下1.9.0版本的torch最后成功了,所以安装1.9.0的Pytorch

Pytorch官方文档:Previous PyTorch Versions | PyTorch

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

安装pytorch3d

这里是一个大坑,我先根据pytorch3d的官方文档直接安装,facebookresearch/pytorch3d: PyTorch3D is FAIR's library of reusable components for deep learning with 3D data (github.com)

不出意料,虽然安装成功,但是最后训练时失败了

原因是:安装pytorch3d时直接采用conda命令在线安装会自动给你安装最新版本的pytorch3d,和作者要求的0.6.0不符合,并且,还会自动给你装别的版本的pytorch(与之前装过的冲突);特别是会多出一个很高版本的torchvision,导致出现a0什么什么的错误,因此选择采用本地安装。

Release Version 0.6.2 · facebookresearch/pytorch3d · GitHub

下载好0.6.0版本后

cd pytorch3d-0.6.2
pip install -e .  

安装MinkowskiEngine

这一步是最坑的,参考博文:

http://t.csdn.cn/pYhZ9

成功运行

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 7
    评论
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值