PyTorch: 从入门到再次入门(一)

Pytorch (以下简称torch)是一个将研究院原型到产品布置无缝衔接的深度学习框架。

torch有两个版本,Staple(1.0)与Preview(Nightly),但作为我们初学者来说,直接安装staple1.0就行了。由于我是以windows入门,所以这篇博客的前一部分就以windows+python系统来进行介绍,后一部分以linux系统来介绍。mac系统或者使用c++可能不适用。

windows下的安装

torch需要较多的python依赖包,所以推荐大家直接安装Anaconda。里面已经将所有相应python包都集成了。(Anaconda安装方法比较简单,可以直接去官网查看就会明白了。但还是不推荐官网下载,太慢了,可以选择一个镜像。我用的是https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ )

安装完anaconda以后,在命令行中写入 conda,如有如下命令,则conda安装完成。

(安装时,不要一直默认安装,可以选择将anaconda直接加入系统路径,就是在安装选择时,将上下两个方框中的全部选中。虽然选中第一个方框时,字体变红且是不推荐的命令,但其实不妨事。)

PyTorch: 从入门到再次入门(一)

初次使用我用的是cpu进行训练,因此可在命令行状态输入:

//为了加速,添加了清华的源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
//安装命令
conda install pytorch-cpu torchvision-cpu

意外错误:

CondaError: Downloaded bytes did not match Content-Length

错误原因:问题如上所示,在用conda安装包的时候下载包的长度不够导致安装包不成功。原因一般是在下载的时候速度较慢,   导致下载timeout而终止。

解决办法:(设置额外的源,如上面代码所述。如果已经换成了清华的源,就不用了)继续键入conda install pytorch-cpu torchvision-cpu,直至安装完成。

然后在命令行的python状态下

import torch
x=torch.rand(5,3)
print(x)

如果有输出结果,则代表安装完成。

PyTorch: 从入门到再次入门(一)

linux安装 

方法基本同上:anaconda+pytorch

安装anaconda:

命令行模式下,执行:

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.3.1-Linux-x86_64.sh
sh Anaconda3-5.3.1-Linux-x86_64.sh

执行完上面命令后,会有一些选项。最后一个问题,即问是否安装microsoft vscode时,选择no,其余选择都选yes。

安装完成后,在命令行状态下输入conda 不报错则安装完成

若是报错,则打开一个新的终端,切换到你原来的用户状态(即如果你安装时是管理员状态,则在运行conda时也要在管理员状态;如果就是自己的用户状态,就在自己的用户状态下运行),输入conda,看是否报错;若不报错,则安装成功

安装pytorch,同上 

//为了加速,添加了清华的源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
//安装命令
conda install pytorch-cpu torchvision-cpu

同样,你是在哪个用户状态下安装的pytorch,也就只能在哪个状态下进行调用(前面有一个base字样,表示当前用户状态是安装pytorch的用户状态) ,其他用户状态可能无法调用。

同样按上面的方法测试,如果可以正确运行,则安装成功。

 

附加知识点

如何在linux下安装jupyter

pip install --upgrade pip
pip install jupyter

我是在root状态下安装的jupyter,所以不能直接jupyter notebook运行jupyter,而是采用下面的命令

jupyter notebook --allow-root //默认端口是8888
jupyter notebook --port 8080 --allow-root //端口被选为8080

然后在浏览器中打开相应链接,便可运行jupyter。