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名字
显示如下图片即可: