1 创建环境
1.1 安装Anaconda
参考本人另一篇博客【Deepin 20系统】Linux系统安装Anaconda和Tensorflow-gpu2.3
1.2 创建一个conda的环境
本人选择的python 版本3.6
conda create -n Pytorch python=3.6
# 环境名称:Pytorch python版本:3.6
进入环境
conda activate Pytorch
2 查看系统Cuda版本
cat /usr/local/cuda/version.txt
3 找到对应版本的执行命令
进入官网选择对应cuda版本的命令。本人的cuda是11的。
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
但是别着急执行命令,因为pytorch有700多M,下载很慢,需要添加镜像后使用
4 添加镜像源
参考conda安装Pytorch下载过慢解决办法(11月26日更新ubuntu下pytorch1.3安装方法)
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
5 执行安装命令
注意:要想通过镜像下载,去掉官网提供的命令的-c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.0