从零开始深度学习Pytorch笔记(1)——安装Pytorch

浏览: 2352

聊到深度学习, 大家第一感觉就是很高大上。就像我们曾经说到机器学习,很多人也是感觉很高大上,但是慢慢接触之后,发现其无非是数学+编程实现,所以从线性回归开始,不断学习,把各种机器学习方法都学习了一遍,并能够通过Python的sklearn库编程实现。有很多朋友和我聊到学习深度学习这个事情,我会推荐他们去看一些相关理论算法,从CNN、RNN到LSTM,从各种传统的深度学习网络结构到比较新的网络结构。在这个过程中发现了一点,就是他们看完了网络结构和算法原理后,往往难以下手去操作,因为Python使用sklearn库实现机器学习的流程非常方便,而且学习了Python的numpy、pandas基础即可快速上手,而深度学习的编程则需要更上一层,对Python新手不太友好,基础部分的张量各种操作和计算图编程都会让新手头大,而深度学习还面临着选择深度学习框架的问题,可供我们选择的框架实在太多,例如较为流行的Tensorflow、Pytorch、Keras、Mxnet、PaddlePaddle等等。各种相关的书也是如雨后春笋般出版出来,我翻看过一些书,还看过一些目录,发现了一个现象,就是各种深度学习框架书在对应的深度学习框架基础部分讲解往往不够详细,让人看完觉得跳跃感很大,而过于强调各种网络结构的实现,这十分不利于新手进行学习。而且初学者会了解到深度学习需要使用GPU训练模型,对于没有GPU配置的初学者,往往觉得学习之路出现了第一道坎,而对于有GPU的初学者,往往因为GPU环境搭建太复杂而从入门到放弃。其实没有GPU环境,也是可以进行深度学习编程学习的。我个人的感受是,GPU环境的搭建变得越来越容易,这是我们放开去学习深度学习的一个好时机,关于在《从零开始深度学习Pytorch笔记》连载中,我会一步步从安装Pytorch环境开始写,慢慢夯实Pytorch的基础,再带大家熟练使用Pytorch编程网络结构,实现深度学习算法。

聊聊为什么使用Pytorch,个人觉得Pytorch比Tensorflow对新手更为友善,大家如果对Tensorflow也感兴趣,完全可以学习了Pytorch之后继续学习Tensorflow。接下来进入正题吧:

如果你没有gpu环境,想使用Pytorch,打开Pytorch的官网:https://pytorch.org/在首页我们可以看到有各种配置可选,我们这里选择CPU的版本(CUDA部分选择None):根据你的操作系统,还有Python的版本可以选择不同选项(当然Python版本问题其实问题不大,因为我们可以通过创建虚拟环境搞定,详情见:你不知道的Python环境管理技巧,超级好用! )选完选项之后,我们看到底部出现了一个命令,我们将命令复制到cmd中运行,即可安装cpu版本的Pytorch(本配置生成的命令为:

pip3 install torch==1.3.1+cpu torchvision==0.4.2+cpu -f https://download.pytorch.org/whl/torch_stable.html

如果下载过慢,可以切换镜像下载,戳这里:Python快速安装库的靠谱办法当然,如果你有GPU环境,或者愿意为学习深度学习去配置GPU,我们可以搭建GPU的Pytorch环境。GPU环境具体搭建流程如下:

(1)安装Python编程环境首先是你得有个Python编程环境,如果没有,推荐安装Anaconda,这个是Python的集成环境,这个Anaconda的安装方法在网络上教程很多(推荐的是对应Python3.6或3.7)。

(2)安装CUDA

我们可以看到,Pytorch对应的CUDA版本为9.2或者10.1:

我们选择CUDA9.2进行下载进入CUDA9.2的下载界面https://developer.nvidia.com/cuda-92-download-archive选择相应的参数后我们点击下载文件:

运行下载好的CUDA安装包:

使用默认路径点击OK安装,一路按照默认安装即可。

安装完成后,测试CUDA是否安装成功,复制安装路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\bin打开cmd,切换到安装路径下,执行: nvcc-V

可以看到CUDA成功安装了。(3)安装cuDNN

cuDNN需要注册登录才能下载,cuDNN下载页面如下:https://developer.nvidia.com/cudnn登陆后,点击Download即可:

然后点击I Agree,点击CUDA9.2,点击win10下载:

之后我们将下载的cuDNN的压缩包解压将解压出的cuda文件夹下的三个文件夹复制到cuda安装目录下,去覆盖掉cuda目录下的三个同名文件夹cuDNN安装完成。
然后进入demo_suite文件夹,查看cuDNN是否安装成功:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\extras\demo_suite切换到该路径下执行bandwidthTest.exe

结果如下图:显示PASS,则cudnn安装成功。我们输入deviceQuery.exe 

可以查询设备信息:最后那段显示的信息:deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 9.2, CUDA Runtime Version = 9.2, NumDevs = 1, Device0 = GeForce GTX 1060 6GB

Result = PASS说明CUDA和cuDNN都安装成功,还显示了我的显卡是GTX 1060 6GB。(4)安装Pytorch

回到Pytorch网站:

https://pytorch.org/我们的选择如下:

可以看到安装命令中有个网站:

https://download.pytorch.org/whl/torch_stable.html

我们Ctrl+F搜索:CU92/torch-1.2.0可以找到我们要的:cu92/torch-1.2.0%2Bcu92-cp37-cp37m-win_amd64.whl然后点击下载即可。然后搜索:torchvision-0.4.0

找到我们要的:cu92/torchvision-0.4.0%2Bcu92-cp37-cp37m-win_amd64.whl点击下载即可。(5)创建项目环境我们打开Python的IDE,例如我喜欢的VS Code,你也可以使用其他软件。

打开后创建一个py文件名为hello pytorch.py(名字任意)。然后在TERMINAL中创建虚拟环境:输入:conda create -n pytorch-gpu python=3.7,并且回车运行关于为什么要创建虚拟环境可以看这里:你不知道的Python环境管理技巧,超级好用!等创建完成后,激活虚拟环境:conda activate pytorch_gpu

然后进入到刚才两个文件的下载目录,我们要在该虚拟环境中安装pytorch我的电脑上目录是:C:\Users\Administrator\Downloads

然后进到目录下pip install 回车进行安装两个文件:我们在py文件中输入:

import torch

print("hello pytorch{}".format(torch.__version__))
print(torch.cuda.is_available())

然后运行代码:结果显示Pytorch安装成功,Pytorch版本为:pytorch1.2.0+cu92CUDA可用!至此,我们的Pytorch GPU版安装成功,如果需要安装过程中的各种资源,可以在公众号回复 “ pytorch安装 ” 获取。

扫码下图关注我们不会让你失望!

image.png

推荐 0
本文由 ID王大伟 创作,采用 知识共享署名-相同方式共享 3.0 中国大陆许可协议 进行许可。
转载、引用前需联系作者,并署名作者且注明文章出处。
本站文章版权归原作者及原出处所有 。内容为作者个人观点, 并不代表本站赞同其观点和对其真实性负责。本站是一个个人学习交流的平台,并不用于任何商业目的,如果有任何问题,请及时联系我们,我们将根据著作权人的要求,立即更正或者删除有关内容。本站拥有对此声明的最终解释权。

0 个评论

要回复文章请先登录注册