安装之间可以先看一下以下教程,方便安装
Linux系统安装Anaconda3保姆级教程
Anaconda创建、删除虚拟环境Linux
我这里就认为你已经安装了anaconda了,安装pytorch之前,我建议先创建一个虚拟环境,在虚拟环境中安装pytorch
首先查看自己的CUDA版本,教程参考Linux 系统查看CUDA和cuDNN版本号
我这里的CUDA版本为9.0.176
随后我们到Pytorch官网,根据自己的情况选择START LOCALLY,它就会自动生成你应该使用的安装命令。如果没有显卡CUDA一栏选择None就可以了。因为我的CUDA版本为9.0,所以我选择cudatoolkit=9.0,并指定了pytorch和torchvision的版本。这里的-c pytorch
意思是选择从pytorch官方源下载,这里不推荐,官方源下载太慢了,还容易莫名其妙的中断,所以这里就去掉了,可看另一篇博客Anaconda换国内源。下载速度会有加成。
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0
OK,输入上面这个命令后我们等就可以了,中间回确认安装,输入y
+ 回车就可以了。
下面这个就是安装成功了,接下来验证一下。
这里引入pytorch的包应输入import torch
# import这个包
import torch
# 这里验证 gpu 可用, True表示可用
torch.cuda.is_available()
妥了!