Win10下pytorch3D安装方法，本人亲测可用

PyTorch3D安装教程：从CUDA到VS2019

原创

已于 2023-06-25 17:57:55 修改 · 3.6k 阅读

22 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #git #github

于 2023-06-16 11:52:27 首次发布

本文详细介绍了如何在拥有GTX1050TiGPU、CUDA11.1、Python3.8和PyTorch1.9.0的环境下，正确安装和配置PyTorch3D0.7.1。关键步骤包括确认版本兼容性、下载CUB和PyTorch3D，安装VisualStudio2019以及使用C++编译源文件。特别强调了环境变量设置和conda环境的使用。

本人的电脑配置如下：

GPU：GTX 1050Ti

CUDA：11.1

python：3.8

pytorch：1.9.0

pytorch3d：0.7.1

NVIDIA CUB：cub-1.9.10-1

1.版本确认

这个步骤非常非常重要，因为不同的pytorch3d版本适配于不同的torch版本，你可以进到facebook/pytorch3d的官网查看不同的适配版本（这是0.7.1的），其余的你们可以自己查看https://github.com/facebookresearch/pytorch3d/releases/tag/v0.7.1

此外，你要保证你的CUDA环境变量要对，比如我的CUDA是11.1，我的cuda toolkit也是11.1的，并把它添加到你的用户的环境变量里，此外还有你的conda也要添加到用户环境变量里面，因为后面我们会用到一个叫做Microsoft Visual Studio2019的工具，在使用时需要你的conda命令进行激活。如果你们有和我一样的版本，cuda11.1，你可以自己新建一个环境，然后重新装一个torch：

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

其它依赖：

最低0.47元/天解锁文章