Pytorch3D安装全流程-亲自安装

简单介绍

Pytorch3D是一个用于加速深度学习在处理3D相关信息时候的运算速率的库。深度学习有时会需要处理大量的3D数据，比如在人脸建模的时候。这时候使用这些库会大大帮助我们节省运算成本。

遇到问题

按照官方的安装方法会遇到不少问题：https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md。我个人在安装的时候，输入相关的指令并不能够成功安装，也是折腾了快一个上午加一个中午才解决，这里分享一下，希望能帮助更多的人。

准备工作

1.conda创建好的虚拟环境

首先你需要一个用conda创建好的虚拟环境，如何创建网上教程很多，在此不多赘述。注意你的cuda里面要有cudatoolkit这个包，这个包是用来使用cuda的，简单来说就是调用你电脑GPU的。

要记好你虚拟环境的python/pytorch版本和cudatoolkit版本，建议pytorch版本手动设定安装版本，否则会自动安装到最新版，最新版往往与别的很多包都是不兼容的

2.cub安装配置

cub的版本一定要和cudatoolkit版本对应，版本对应说明的链接在这里：https://gitcode.net/mirrors/nvidia/cub?utm_source=csdn_github_accelerator

知道自己版本是什么了之后，开始下载，下载链接：https://github.com/NVIDIA/cub/releases