参考其他博主的文章按照官网对应版本安装和下载。如下是cuda、cuDNN、python、pytorch、torchvision、CUB、pytorch3d的对应关系表
这是pytorch3d官网的版本对应声明:其实没啥用
然后是我参考的安装教程如下:
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。本文链接:https://blog.csdn.net/m0_70229101/article/details/127196699
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。本文链接:https://blog.csdn.net/matt45m/article/details/133896773
问题来了:
在最后编译pytoch3d的时候,死都装不上,一直报错:error: command 'C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.6\\bin\\nvcc.exe' failed with exit code 1
解决步骤:
放弃按照官网的方式,而是用git命令
首先安装git,下载链接:git官网
然后桌面右键git bash,输入如下命令并执行:
git clone https://github.com/facebookresearch/pytorch3d.git
会自动匹配版本,在桌面下载pytorch3d
再将下载好的文件放到conda的虚拟环境中,一般是 .\envs\PCN\Lib\site-packages
这里注意下,我参考一篇文章,下载并安装了MinGW,这个不确定是否需要,最好装一下
再然后激活环境,因为我用miniconda的,所以直接用Anaconda Prompt (miniconda3)的控制台激活环境,而不是像其他文章一样用x64 Native Tools Command Prompt for VS 2019。
接着用cd命令定位到pytorch3d的目录下,再就是和参考文章一样,修改setup.py中的编译器设置代码,执行
set DISTUTILS_USE_SDK=1
set PYTORCH3D_NO_NINJA=1
python setup.py install
等待十几分钟左右就会编译完成,会显示(忘记截图了,借用别人装好的图)
最后自己import pytorch3d试试,是否成功导入,测试代码参考文章
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。本文链接:https://blog.csdn.net/yangyu0515/article/details/131910664
补充一下我的环境:GPU 4060,CUDA 11.6,python 3.8,pytorch 1.13,torchvision 0.14,VS 2019,CUB 1.5,pytorch3d 0.7.7
没错给我匹配个最新版的0.7.7,关键还能装上去
至于VS多版本存在是否影响编译,这个我没试过,我是把2022卸载了,留了个2019