安装好anaconda后,安装pytorch:

(1)使用pip install** ,代替conda install***更方便

(2)需要用到GPU资源的话,要加cuda的内容。linux查看本机cudu版本的命令为:cat /usr/local/cuda/version.txt

(3)知道了cuda的版本,去查看对应torch和torchvision的版本

Linux pytorch安装 cuda,torch,torchvision版本选择_torchLinux pytorch安装 cuda,torch,torchvision版本选择_torchvision版本_02

 

 我的cuda版本是10.0.130,所以我要安装的pytorch版本是1.1,1.2这些。我选择1.2版本,那torch对应也是1.2版本,torchvision就选择0.4.0版本。

(4)直接在shell里输入:pip install torch==1.2.0 torchvison==0.4.0,回车,等他运行结束

(5)shell键入“python”进入python环境,然后输入“import torch",看看是否成功,”print(torch.cuda.is_available()) ”查看cuda版本是否对应。如果没有按照上面的版本对应的话,他会返回“False”,对应上了,就返回“True"。