最近看到这篇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
这一步是最坑的,参考博文: