Ubuntu安装Anaconda并设置pytorch与pytorch3D环境

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

前言

安装pytorch与pytorch3d环境

---

一、pytorch与pytorch3d的关系

PyTorch3D是由Facebook AI Research (FAIR)团队开发的一个用于3D数据深度学习的库。它提供了高效、可重用的组件，旨在加速3D计算机视觉研究。PyTorch3D主要用于处理三角网格、点云和体积数据，并支持可微渲染。

PyTorch3D需要pytorch库支持，因此，PyTorch3D的版本更新速度慢于pytorch，因此，需要选择并安装PyTorch3D支持的pytorch库。

二、pytorch安装

1. 查看pytorch3d的最新版本

进入网页pytorch3d离线安装包，查看最新的pytorch3d库支持的pytorch版本，并点击下载该版本。

例如：pytorch3d-0.7.8-py39_cu121_pyt241.tar.bz2

其中，py39表示python版本需为3.9，cu121表示cuda版本为12.1（cuda是向下兼容的，cuda版本高于12.1均可），pyt241表示pytorch版本为2.4.1。

2.安装pytorch

打开terminal

新建pytorch3d环境，并指定python版本为3.9

conda create -n pytorch3d python=3.9 -y
conda activate pytorch3d

进入pytorch历史版本网页，选择上一步确定好的各个版本(cu121表示cuda版本为12.1，pyt241表示pytorch版本为2.4.1)，拷贝pytorch的安装代码并在terminal中运行。

conda install pytorch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 pytorch-cuda=12.1 -c pytorch -c nvidia -y

二、pytorch3d安装

pytorch3安装官方流程

安装依赖库

conda install -c iopath iopath -y

# Demos and examples
conda install jupyter -y
pip install scikit-image matplotlib imageio plotly opencv-python

# Tests/Linting
conda install -c fvcore -c conda-forge fvcore -y
pip install black usort flake8 flake8-bugbear flake8-comprehensions

进入保存pytorch3d离线安装包的文件夹，运行：

conda install pytorch3d-0.7.8-py39_cu121_pyt241.tar.bz2

---