【Deepin 20系统】Linux系统Anaconda安装Pytorch

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