PyTorch3D 安装记录

小猿805

已于 2023-06-15 13:25:10 修改

阅读量5.5k

点赞数 9

文章标签： pytorch 3d 深度学习

于 2023-06-14 17:11:54 首次发布

本文链接：https://blog.csdn.net/weixin_43891061/article/details/131211465

版权

文章提供了PyTorch和PyTorch3D的安装步骤，强调了CUDA和Python版本的匹配，给出了conda和pip安装命令，并提供了不同CUDA和PyTorch版本对应的PyTorch3D安装链接。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. PyTorch 安装

PyTorch 官网：PyTorch 官网

最新版本的 PyTorch：最新版本的 PyTorch

以 cuda 11.3，Python 3.9，Pytorch 1.12.1 为例

# conda 安装
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
# pip 安装
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

2. PyTorch3D 安装

直接使用 pip 安装 PyTorch3D 会提示错误

在这里插入图片描述

进入 Anaconda 官网找到 PyTorch3D：PyTorch3D Anaconda 官网

找到和 cuda 以及 Python 版本对应的 Pytorch3D 版本，找到正确的版本后，右键绿色的文件名选择复制链接地址

3. 一些常用的 PyTorch3D 链接

使用 conda instasll PyTorch3D 链接即可，

如安装对应 Python 3.9、cuda 11.3 Pytorch 1.12.1 版本的 PyTorch3D，使用如下命令：

conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.4/download/linux-64/pytorch3d-0.7.4-py39_cu113_pyt1121.tar.bz2

3.1 cuda 11.3 Python 3.8

1. cuda 11.3 Python 3.8 PyTorch 1.12.0 对应 pytorch3d-0.7.4-py38_cu113_pyt1120

https://anaconda.org/pytorch3d/pytorch3d/0.7.4/download/linux-64/pytorch3d-0.7.4-py38_cu113_pyt1120.tar.bz2

安装命令：

conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.4/download/linux-64/pytorch3d-0.7.4-py38_cu113_pyt1120.tar.bz2

2. cuda 11.3 Python 3.8 PyTorch 1.12.1 对应 pytorch3d-0.7.4-py38_cu113_pyt1121

https://anaconda.org/pytorch3d/pytorch3d/0.7.4/download/linux-64/pytorch3d-0.7.4-py38_cu113_pyt1121.tar.bz2

安装命令：

conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.4/download/linux-64/pytorch3d-0.7.4-py38_cu113_pyt1121.tar.bz2

3.2 cuda 11.3 Python 3.9

1. cuda 11.3 Python 3.9 Torch 1.12.0 对应 pytorch3d-0.7.4-py39_cu113_pyt1120

https://anaconda.org/pytorch3d/pytorch3d/0.7.4/download/linux-64/pytorch3d-0.7.4-py39_cu113_pyt1120.tar.bz2

安装命令：

conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.4/download/linux-64/pytorch3d-0.7.4-py39_cu113_pyt1120.tar.bz2

2. cuda 11.3 Python 3.9 Torch 1.12.1 对应 pytorch3d-0.7.4-py39_cu113_pyt1121

https://anaconda.org/pytorch3d/pytorch3d/0.7.4/download/linux-64/pytorch3d-0.7.4-py39_cu113_pyt1121.tar.bz2

安装命令：

conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.4/download/linux-64/pytorch3d-0.7.4-py39_cu113_pyt1121.tar.bz2

3.3 cuda 11.7 Python 3.9

1. cuda 11.7 Python 3.9 Torch 2.0.1 对应 pytorch3d-0.7.4-py39_cu117_pyt201

https://anaconda.org/pytorch3d/pytorch3d/0.7.4/download/linux-64/pytorch3d-0.7.4-py39_cu117_pyt201.tar.bz2

安装命令：

conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.4/download/linux-64/pytorch3d-0.7.4-py39_cu117_pyt201.tar.bz2

4. 安装完成

使用 conda list 查看 PyTorch3D 是否安装成功
在这里插入图片描述