Deepin20.2.4安装Pytorch GPU版本

Deepin20.2.4安装Pytorch GPU版本

按照前文安装好Nvidia驱动、CUDA、CUDNN。
打开anaconda官网下载最新版的anaconda(linux版)，在下载的目录执行：

bash Anaconda3-2021.05-Linux-x86_64.sh

中途可以一直回车/yes，即可安装成功。
这时我们需要设置环境变量，命令行输入：

sudo vim /etc/profile

在文件末尾加上如下代码：

#Anaconda
export PATH=$PATH:/home/zq1993/anaconda3/bin

保存之后输入source /etc/profile使其生效。
接着命令行输入：

anaconda-navigator

启动anaconda图形化界面。
在命令行启动命令，新建一个新的虚拟环境设置python版本为3.6。

conda create -n py3.6 python=3.6

接着激活这个虚拟环境：

conda activate py3.6

这个时候我们需要去下载pytorch的离线文件在线安装总是出现各种奇葩问题。
根据我们前文的情况(CUDA版本11.1)来下载如下文件：

torch：

torchvision

porch和porchvision是有对应关系的！
这里显示了torch和torchvision之间版本的对应关系。

接着使用pip进行离线安装。（这里用pip不要用conda)

pip install torch-1.10.0+cu111-cp36-cp36m-linux_x86_64.whl 

pip install torchvision-0.11.1+cu111-cp36-cp36m-linux_x86_64.whl -i https://pypi.douban.com/simple/

第二段代码加入豆瓣源方便快速下载(因为新建的虚拟环境需要加载第三方包，不建议直接修改anaconda的源，有些源不稳定可能哪一天下载包就会失联)。
这个时候我们来测试一下是否可以使用：

import torch
import torchvision

再输入：

torch.cuda.is_available()
torch.cuda.device_count()#gpu数量
torch.cuda.get_device_name(0)#gpu名字

显示如下图片即可：