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

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

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

前言

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

注:安装环境的时候,如果不想自己踩坑,最好是按着博客给的配置一模一样的安装,如果你喜欢自己解决各种报错,踩各种莫名其妙的坑,那环境随便装,配置随便改,上面的话当我没说,环境配置: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就可以解决这个错误。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

知来者逆

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值