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

原创已于 2024-03-18 20:36:18 修改 · 2.3k 阅读

19 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #人工智能 #pytotch3d #数字人

于 2023-10-19 10:38:16 首次发布

数字人专栏收录该内容

24 篇文章

订阅专栏

PyTorch3D是一个基于PyTorch的开源库，专为3D计算机视觉提供工具，支持3D几何操作、渲染和数据处理。文章介绍了其主要功能、用途以及Windows环境下安装步骤，包括官方和源码安装方法及常见错误解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前言

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

PyTorch3D的用途：

3D 几何操作：PyTorch3D 提供了一系列用于处理 3D 几何数据的工具和函数。它允许用户进行点云操作、模型旋转、平移、缩放、重采样等。
3D 网格操作：该库支持处理 3D 网格数据，用户可以执行网格变换、网格采样、法线估计、网格融合等操作。
3D 渲染：PyTorch3D 提供了 3D 渲染引擎，允许用户在 3D 场景中渲染图像、视角的变换、光照和阴影等操作。
3D 数据加载和处理：用户可以使用 PyTorch3D 加载和处理不同格式的 3D 数据，包括 3D 模型、点云、深度图等。
PyTorch 集成：由于 PyTorch3D 是基于 PyTorch 开发的，因此用户可以利用 PyTorch 的自动微分机制进行梯度计算，便于深度学习任务的建模和训练。

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

安装Pytorch3D

注：安装环境的时候，如果不想自己踩坑，最好是按着博客给的配置一模一样的安装，如果你喜欢自己解决各种报错，踩各种莫名其妙的坑，那环境随便装，配置随便改，上面的话当我没说，环境配置：win 10, N卡，cuda 11.7, cudnn 8.5,Anaconda3,VS2019

安装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/18kIVsROuycBN9dlZShuCZw
提取码：1dz3

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就可以解决这个错误。