在Win10安装pytorch3d

Pytorch3d

这是一个为使用Pytorch的3D计算机视觉研究提供的高效可重用的组件

主要特点包括：

1. 用于储存和操作三角形网格的数据结构
2. 三角形网格的高效操作（投影变换、图形卷积、采样、损失函数）
3. 可区分的网格渲染器
4. 可以通过隐式函数进行新试图合成的框架

由于pytorch3d旨在用于预测和操作3D数据的深度学习方法的顺利集成，因此在其中的运算符：

• 使用pytorch张量实现
• 可以处理小批量的异构数据
• 可区分
• 可以使用GPU加速

以上是对于这个包的简要介绍，需要了解更多，可以点击官方链接

以下操作都是在anaconda虚拟环境中进行的，如果有小伙伴想要建立虚拟环境可以参考：链接

安装pytorch3d

首先需要检查自己的cuda以及pytorch版本

1.安装torch和Cuda

cuda安装地址：https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local

如果已经装好，检查自己的torch和cuda版本的是否可用

import torch
torch.cuda.is_available()

如果返回是True则已经在使用GPU版本的torch

官网对比版本查询

如果返回不是True，则需要卸载重新安装

pip uninstall torch torchvision torchaudio
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

最后再用**torch.cuda.is_available()**检查即可

2.安装Pytorch3d

1.conda install -c fvcore -c iopath -c conda-forge fvcore iopath

2.进入下载地址，在最下面选择自己选哟的版本

3.再下拉到最下面进行下载

4.解压后在文件夹中

python setup.py install

此时import pytorch3d 也是ok的