(python篇）安装pytorch+pytorch3d教程

# 前言

安装pytoch以及pytorch3d最重要的就是弄清两件事情：python版本+cuda版本

# 具体安装操作

1. 创建虚拟环境

conda create -n your_env_name python=3.X -y

当你创建虚拟环境时你需要确认python版本，目前通常使用3.8或者3.9，这两个版本能够更容易找到对应版本的pytoch和pytorch3d。

2. 确定当前环境中存在的cuda版本

通常我们使用“nvidia-smi”查看的版本CUDA Version只是兼容的驱动支持的 CUDA 版本，并不是你具体安装的CUDA版本，你具体安装的版本只要在这个版本以下就能兼容。

3. 安装pytorch

官方链接：Get Started

常用的有两种方式，第一种是选择 【Pip】直接线上安装，第二种则是选择【LibTorch】先将包离线下载下来然后再安装。我在下面提供几个版本的pyroch安装：

## pytorch1.13.1+cuda11.6
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

## pytorch2.2.0+cuda11.8
conda install pytorch==2.2.0 torpythochvision==0.17.0 torchaudio==2.2.0 pytorch-cuda=11.8 -c pytorch -c nvidia
pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu118

## pytorch2.3.0+cuda12.1
conda install pytorch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 pytorch-cuda=12.1 -c pytorch -c nvidia
pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121

4. 安装pytorch3d

安装pytorch3d更容易出错，所以我更喜欢用离线下载的方式，确保不会出错。

（1）在官网中寻找你对应版本的pytorch3d，你需要确定【cuda版本】+【python版本】+【pytorch版本】

官方链接：https://anaconda.org/pytorch3d/pytorch3d/files?page=2

（2）下载后你会得到一个xxx.tar.bz2，然后使用命令（注意要cd到压缩包对应的位置，通常就放在/home的位置就可以）：

conda install pytorch3d-0.7.8-py39_cu118_pyt220.tar.bz2