Windows平台下安装pytorch3D

参考其他博主的文章按照官网对应版本安装和下载。如下是cuda、cuDNN、python、pytorch、torchvision、CUB、pytorch3d的对应关系表

e220d6c7ab25435f8f7f4f5eabfaf3d2.png

300e95b36b84449e887b9b29bff20a2f.png

f354c02a7ee245cca96f7e7e56c97f77.png

这是pytorch3d官网的版本对应声明:其实没啥用

4174718732904a2690a88634c79a2ac9.png

然后是我参考的安装教程如下:

版权声明:本文为博主原创文章,遵循 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

等待十几分钟左右就会编译完成,会显示(忘记截图了,借用别人装好的图)

5f880987970342178e80f44f517d6e42.png 最后自己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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值