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
安装命令:文章来源:https://www.toymoban.com/news/detail-632343.html
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 是否安装成功
文章来源地址https://www.toymoban.com/news/detail-632343.html
到了这里,关于PyTorch3D 安装记录的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!