通过Anaconda来安装
-
- 配置好清华镜像或者其他镜像
- 设置延时参数以应对网速慢的情况
- 新建虚拟环境
-
确定目标的pytorch版本
- 从latest-version 或 previous-versions中选择需要的pytorch版本
- 同时确定其他重要组件的版本
- torchvision
- torchaudio
- cudatoolkit-version
下载地址:https://pytorch.org/get-started/locally/ or https://pytorch.org/get-started/previous-versions/
-
安装方式
切换到对应的虚拟环境
source/conda activate {env-name}
-
第一种,直接通过torch官方指导的命令行进行直接安装,conda会直接指定适配的版本。
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
- 网速稳定快速,能够顺利安装
- 该命令会将所有需要的库下载到路径:~/anaconda3/pkgs下,并逐一进行安装
-
手动本地安装
-
特殊版本的下载,通过Chrome或wget命令
-
清华镜像的torch库,下载pytorch,torchvision,torchaudio
-
注意选择版本,包括对应的python、cuda、cudnn版本,如
- pytorch-1.9.0-py3.9_cuda11.1_cudnn8.0.5_0.tar.bz2
- torchaudio-0.9.0-py38.tar.bz2
- torchvision-0.9.0-py38_cu111.tar.bz2
清华镜像pytorch库地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
-
-
cudatoolkit下载 -- 版本与CUDA Runtime Version对应
- 网址 + {package-version} + {-} + {build-version} + {.tar.bz2}
- 网址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/linux-64/
- 这些大文件在下载之前可以去访问一下服务器的公共资源库,看一下有没有,可以避免重复下载.
-
-
手动安装大文件
conda install --offline {root}/{pkg-name} #
成功解压和安装,会将解压的文件保存在~/anaconda3/pkgs/{pkg-name}下
-
剩余文件安装
-
使用Pytorch官网推荐的命令,如
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
-
-
-
-
卸载与重新安装
如果遇到pytorch版本安装错误时,可以使用以下命令卸载已安装的pytorch
source activate {env-name} conda uninstall pytorch