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直接加入系统路径,就是在安装选择时,将上下两个方框中的全部选中。虽然选中第一个方框时,字体变红且是不推荐的命令,但其实不妨事。)
初次使用我用的是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)
如果有输出结果,则代表安装完成。
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。