liunx+cuda8.0 安装pytorch 1.0.1和torchvision 0.2.1

1.准备

首先需要安装anaconda,本文不介绍安装过程。

2.创建python虚拟环境

我们使用 conda 创建名为 mypytorch 的虚拟环境，并且 python 版本为3.7

在命令行输入命令：conda create -n mypytorch python=3.7

其中-n后面的是虚拟环境名，可以自行输入，python的版本也可以根据实际情况输入

 

根据提示是否安装这些包，我们输入 y 进行安装

 

出现下面代码说明环境创建成功

 

3.进入自己创建的环境

输入命令：conda activate mypytroch

回车后，前面的括号变成自己的环境名说明已经进入自己的虚拟环境

 

4.安装pytorch 1.0.1

（1）首先查看操作系统信息

输入命令：uname -a

可以看到我们的是64位系统，这个后面装pytorch 有用！！

（2）查看cuda版本

输入命令：cat /usr/local/cuda/version.txt

注意 cat 后面有空格，不然查不到！！

这里我们的 cuda 版本是8.0

 

（3）下载对应pytorch的whl文件

进入pytorch官网：https://pytorch.org/

因为我们的cuda版本很低，所以我们需要下载对应的pytorch.whl文件

进入官网，找到 previous versions of Pytorch

然后找到cuda8.0版本的pytorch列表

找到自己需要的pytorch版本，必须和自己的python版本，cuda版本，以及操作系统对应

我们这里选择的是 torch1.0.1 ,  python3.7  以及 64位的liunx

 

 

（4）文件上传至linux服务器

我用的上传软件是FileZilla

（5）使用命令行进入该文件路径

用 cd 命令进入文件夹 

（6）安装该whl文件

在文件路径下使用命令： pip install +文件全名

注意：必须用 pip,不能用conda,需要在自己的虚拟环境下

提示如上，说明安装成功。

（7）检查是否安装成功

提示：因为torch需要numpy的支持，所以需要检查是否安装了numpy,如果没有安装，则使用命令 pip install numpy 安装

输入python 进入python 环境

输入命令：import torch

                  print(torch.cuda.is_available())

如果没报错，说明pytorch安装成功，如果返回的是True,说明当前的pytorch可以使用cuda

 

5.安装torchvision 0.2.1

因为torchvison 和 torch 版本需对应，而我们装的torch是1.0.1,所以我们这里选择安装torchvison 0.2.1

 

 

使用命令 pip install torchvision==0.2.1

 

显示安装成功，大功告成！！！！