首先感谢前人的经验文章,使我少走不少弯路。最近三维视觉的文章越来越多,CVPR2022的三维视觉文章已经有了很大的比重。而最近关于点云的文章越来越多的使用pytorch3D来直接训练,要复现这些文章,无疑pytorch3D更加方便。
参考这篇文章以及pytorch3D的指引,我们首先使用anaconda3新建一个虚拟环境:
conda create -n pytorch3d python=3.8
conda activate pytorch3d
然后安装对应版本的包:
conda install -c pytorch pytorch=1.7.1 torchvision=0.8.2 cudatoolkit=10.2
其中,cudatoolkit版本要和自己的cuda版本一致。
然后进入地址,下载1.10.0的cub版本,解压,在电脑的系统变量中创建 CUB_HOME,将其值设为 你的解压地址/cub-1.10.0