windows 10下安装Pytorch的操作文档与避坑指南

前言

PyTorch3D 是 Facebook AI Research (FAIR) 开发的一个用于 3D 计算机视觉任务的开源深度学习库。它是建立在 PyTorch 框架之上的，并旨在为研究人员和工程师提供处理 3D 数据、渲染和几何计算的工具和模块。

PyTorch3D的用途：

1. 3D 几何操作：PyTorch3D 提供了一系列用于处理 3D 几何数据的工具和函数。它允许用户进行点云操作、模型旋转、平移、缩放、重采样等。

2. 3D 网格操作：该库支持处理 3D 网格数据，用户可以执行网格变换、网格采样、法线估计、网格融合等操作。

3. 3D 渲染：PyTorch3D 提供了 3D 渲染引擎，允许用户在 3D 场景中渲染图像、视角的变换、光照和阴影等操作。

4. 3D 数据加载和处理：用户可以使用 PyTorch3D 加载和处理不同格式的 3D 数据，包括 3D 模型、点云、深度图等。

5. PyTorch 集成：由于 PyTorch3D 是基于 PyTorch 开发的，因此用户可以利用 PyTorch 的自动微分机制进行梯度计算，便于深度学习任务的建模和训练。

PyTorch3D 的目标是为研究人员和开发者提供强大的 3D 计算机视觉工具，使其能够更容易地进行 3D 图像处理、渲染、几何计算以及深度学习模型的设计和训练。它适用于各种应用领域，包括计算机图形学、计算机视觉、机器学习和人工智能。你可以在 GitHub 上找到 PyTorch3D 的开源代码和文档，以深入了解其功能和用法。

安装Pytorch3D

安装pytorch3d，官方的要求是要有C++编译器：

gcc & g++ ≥ 4.9

那在在windows中，可以安装Visual Studio 来进行编译，我这里使用的是Visual Studio 2019 企业版本，安装时注意勾选下面的选项，安装完成之后重启电脑，如果在编译过程中提示找不Visual Studio，看下环境变量是否加上了Visual Studio。

1.官网安装方式

conda create -n pytorch3d python=3.9
conda activate pytorch3d
conda install pytorch=1.13.0 torchvision pytorch-cuda=11.6 -c pytorch -c nvidia
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
pip install "git+https://github.com/facebookresearch/pytorch3d.git"

出现以下提示代表安装成功：

2.源码安装

下载pytorch3d源码,如果下载不了，按上面的百度网盘下载：https://pan.baidu.com/s/178WbPBRtCl3hF8b3niCeSA 提取码：icd8

git clone https://github.com/facebookresearch/pytorch3d.git
cd pytorch3d
python setup.py install

在安装pytorch3d可能出现错误，可以看文章结尾列的常见错误。

3.常见错误

subprocess.CalledProcessError: Command ‘[‘ninja’, ‘-v’]’ returned non-zero exit status 1.

出现这个错误时，找到dist-packages/torch/utils/cpp_extension.py这个文件，找到command = [‘ninja’, ‘-v’]，改成 command = [‘ninja’, ‘–version’]，改了效果如下：

   # command = ['ninja', '-v']
    command = ['ninja', '--version']

AttributeError: ‘Upsample’ object has no attribute ‘recompute_scale_factor’

把cuda改成11.7就可以解决这个错误。