1. PyTorch 安装
-
PyTorch 官网:PyTorch 官网
-
最新版本的 PyTorch:最新版本的 PyTorch
以 cuda 11.1,Python 3.8,Pytorch 1.8.0 为例
# conda 安装
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
# pip 安装
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
2. PyTorch3D 安装
- 由于直接使用 pip 进行PyTorch3D 安装会报错,所以选择在官网直接下载对应的安装包进行安装。
- 进入 Anaconda 官网找到 PyTorch3D:PyTorch3D 官网
找到和 cuda 以及 Python 版本对应的 Pytorch3D 版本,翻页找到正确的版本后,右键绿色的文件名选择复制链接地址。如果没有找到对应你环境的版本,也可以重新创建一个对应的环境,再进行pytorch和pytorch3D的安装即可。
2.1 安装PyTorch3D
使用 conda install
+ 复制的 PyTorch3D 链接
如 安装对应 Python 3.8 cuda 11.1 Pytorch 1.8.0 版本的 PyTorch3D,使用如下命令:
conda install https://anaconda.org/pytorch3d/pytorch3d/0.7.0/download/linux-64/pytorch3d-0.7.0-py38_cu111_pyt180.tar.bz2
2.2 查看是否安装成功
conda list
成功安装!
参考:
PyTorch3D 安装记录