PyTorch3D 安装记录_pytorch3d官网

作者：盐析白兔 | 2024-03-25 08:55:08

踩

pytorch3d官网

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
1
2
3
4

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
1

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
1

安装命令：

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

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
1

安装命令：

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

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
1

安装命令：

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

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
1

安装命令：

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

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
1

安装命令：

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

4. 安装完成

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

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/308428