PyTorch的深度学习入门之PyTorch安装和配置


Posted in Python onJune 27, 2019

前言

深度神经网络是一种目前被广泛使用的工具,可以用于图像识别、分类,物体检测,机器翻译等等。深度学习(DeepLearning)是一种学习神经网络各种参数的方法。因此,我们将要介绍的深度学习,指的是构建神经网络结构,并且运用各种深度学习算法训练网络参数,进而解决各种任务。本文从PyTorch环境配置开始。PyTorch是一种Python接口的深度学习框架,使用灵活,学习方便。还有其他主流的深度学习框架,例如Caffe,TensorFlow,CNTK等等,各有千秋。笔者认为,初期学习还是选择一种入门,不要期望全都学会。须知,发力集中才能深入挖掘。乱花渐欲迷人眼,选择适合自己的,从一而终,相信会对科研大有裨益!

***************************************************

说明:本文乃至本系列全部文章都是在PyTorch0.2版本下做的。现在版本更新了很多,会有一些API的更改,请大家注意Follow最新的变化,以免由于版本问题受到困扰。

***************************************************

1. 环境说明

PyTorch目前支持OSX和Linux两种系统,并且支持多种安装方式。在官网上介绍了基于conda,pip和源代码编译几种不同的安装方式。支持的Python版本有2.7,3.5和3.6。鉴于深度学习需要的计算量一般比较大,强烈建议找到一个有独立显卡的电脑来展开学习,当然没有显卡也能用,就是计算慢很多了。

如果你的电脑是Windows,那么可以安装一个虚拟机来运行Linux,但是性能可能会折扣。我的电脑操作系统是Ubuntu16.04,所以我会以此为例子来介绍后面的内容。

PyTorch的深度学习入门之PyTorch安装和配置

 2. Anaconda 和Python

PackageManager我们选择conda,于是我们需要安装Anaconda这个功能强大的包,下载地质:https://www.anaconda.com/download/ 。里面包含了conda工具,也有Python,以及很多Python需要的扩展工具包。选择2.7版本,下载并安装即可。接下来,如果有显卡,进入第3步;没有显卡直接到第4步。

PyTorch的深度学习入门之PyTorch安装和配置

3. 显卡驱动和CUDA

要使用显卡进行运算,你需要使用支持CUDA的NVIDIA显卡,目前比较好的显卡有NVIDIATITANX、GTX1080Ti 等。好的显卡将会是深度学习研究的有力武器。当然,普通的显卡例如GTX970、GTX1060等也是可以用的。实在没有显卡,那只能做简单的小数据量的实验,效果不会太好。

在Ubuntu16.04上可以采用以下方法安装显卡驱动,这种方式比较稳定。打开“SystemSettings”——“Software&Updates” —— “AdditionalDrivers”,联网状态会自动搜索可用的显卡驱动,选择可用的版本,点击“ApplyChanges”即可。可能重启动后显卡驱动才能生效。

接下来安装CUDAToolkit。因为最新的CUDA是9.0版本,但是PyTorch只能支持到8.0。所以转到该网址“https://developer.nvidia.com/cuda-80-ga2-download-archive”,选择对应的参数,下方会有对应的CUDA文件。例如,下图是我选择的版本。选择deb文件,发现有1.9G大小,下载下来,按照baseinstaller的指示来完成安装。附加的cuBLAS也可以下载下来安装上。

CUDA安装完成后,在主目录下打开“.bachrc”文件,在末尾添加如下代码:

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

 PyTorch的深度学习入门之PyTorch安装和配置

4. 运行PyTorch的安装命令

如果你和我一样,采用8.0CUDA,2.7版本Python,可以运行以下命令:

conda install pytorch torchvision cuda80 -c soumith

如果没有显卡,2.7版本Python,可以这样:

conda install pytorch torchvision -c soumith

5. 验证安装是否成功

要显示显卡信息,在终端输入:

sudo lshw -c video

要显示CUDA信息,在终端输入:

nvcc -V

查看Phthon版本,在终端输入:

python --version

验证pytorch是否安装成功,在终端输入:

python

此时进入python环境。然后,继续输入

import torch

import torchvision

不报错就表明安装成功。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
使用python实现递归版汉诺塔示例(汉诺塔递归算法)
Apr 08 Python
Python手机号码归属地查询代码
May 04 Python
Python的几个高级语法概念浅析(lambda表达式闭包装饰器)
May 28 Python
Sanic框架异常处理与中间件操作实例分析
Jul 16 Python
对Python3 goto 语句的使用方法详解
Feb 16 Python
python实现AES加密和解密
Mar 27 Python
浅谈Python大神都是这样处理XML文件的
May 31 Python
Python pandas RFM模型应用实例详解
Nov 20 Python
python GUI库图形界面开发之PyQt5图片显示控件QPixmap详细使用方法与实例
Feb 27 Python
Django权限设置及验证方式
May 13 Python
Python用来做Web开发的优势有哪些
Aug 05 Python
PyCharm2020.3.2安装超详细教程
Feb 08 Python
解决pycharm remote deployment 配置的问题
Jun 27 #Python
python turtle库画一个方格和圆实例
Jun 27 #Python
Python实现的对一个数进行因式分解操作示例
Jun 27 #Python
python pytest进阶之xunit fixture详解
Jun 27 #Python
Python批量查询关键词微信指数实例方法
Jun 27 #Python
Django框架orM与自定义SQL语句混合事务控制操作
Jun 27 #Python
Pycharm保存不能自动同步到远程服务器的解决方法
Jun 27 #Python
You might like
php的控制语句
2006/10/09 PHP
MYSQL环境变量设置方法
2007/01/15 PHP
dedecms防止FCK乱格式化你的代码的修改方法
2007/03/17 PHP
PHP字符串处理的10个简单方法
2010/06/30 PHP
用 Composer构建自己的 PHP 框架之使用 ORM
2014/10/30 PHP
php比较两个字符串长度的方法
2015/07/13 PHP
PHP使用缓存即时输出内容(output buffering)的方法
2015/08/03 PHP
php-beanstalkd消息队列类实例分享
2017/07/19 PHP
TP(thinkPHP)框架多层控制器和多级控制器的使用示例
2018/06/13 PHP
JavaScript的public、private和privileged模式
2009/12/28 Javascript
JavaScript高级程序设计 读书笔记之十一 内置对象Global
2012/03/07 Javascript
单击按钮显示隐藏子菜单经典案例
2013/01/04 Javascript
有关于JS构造函数的重载和工厂方法
2013/04/07 Javascript
使用ImageMagick进行图片缩放、合成与裁剪(js+python)
2013/09/16 Javascript
在其他地方你学不到的jQuery小贴士和技巧(欢迎收藏)
2016/01/20 Javascript
jQuery ajax提交Form表单实例(附demo源码)
2016/04/06 Javascript
JavaScript数据类型转换的注意事项
2016/07/31 Javascript
jQuery实现鼠标经过像翻页和描点链接效果
2016/08/08 Javascript
原生js实现日期计算器功能
2017/02/17 Javascript
vue实现点击当前标签高亮效果【推荐】
2018/06/22 Javascript
详解nuxt sass全局变量(公共scss解决方案)
2018/06/27 Javascript
vue-cli项目修改文件热重载失效的解决方法
2018/09/19 Javascript
layui table去掉右侧滑动条的实现方法
2019/09/05 Javascript
python 生成不重复的随机数的代码
2011/05/15 Python
python控制台显示时钟的示例
2014/02/24 Python
Python实现TCP/IP协议下的端口转发及重定向示例
2016/06/14 Python
python读取一个目录下所有txt里面的内容方法
2018/06/23 Python
Flask框架web开发之零基础入门
2018/12/10 Python
如何基于Python爬取隐秘的角落评论
2020/07/02 Python
pytorch 计算Parameter和FLOP的操作
2021/03/04 Python
网络信息管理员岗位职责
2014/01/05 职场文书
迎新春趣味活动方案
2014/08/24 职场文书
老人节标语大全
2014/10/08 职场文书
党支部书记岗位职责
2015/02/15 职场文书
2019大学生暑期实习心得总结
2019/08/21 职场文书
Django使用redis配置缓存的方法
2021/06/01 Redis