Linux 安装pytorch3d

L_e_e_

已于 2023-05-15 13:13:22 修改

阅读量1.2k

点赞数 2

于 2023-05-09 17:29:17 首次发布

本文链接：https://blog.csdn.net/L_e_e_/article/details/130583781

版权

本文详细记录了如何在Linux环境下使用conda创建新环境，并安装PyTorch1.13.0、torchvision、CUDA11.6以及相关的库如fvcore、iopath、bottleneck和nvidia-cub，最终成功安装PyTorch3D。按照官方教程操作，即使提供的命令不全，也能确保安装过程顺利进行。

摘要由CSDN通过智能技术生成

本文主要记录在Linux上安装pytorch3d的步骤，其实按照官网教程就可以安装成功。
Install PyTorch3D

首先创建并进入一个新的环境：

conda create -n XXXX python=3.9
conda activate XXXX

然后依次输入以下命令：

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

conda install -c bottler nvidiacub

conda install pytorch3d -c pytorch3d

实测可以跑通。

注：上面的命令并不完整，但是对我来说已经足够，建议按官网的教程操作