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 28 Python
Python2.7下安装Scrapy框架步骤教程
Dec 22 Python
python 通过字符串调用对象属性或方法的实例讲解
Apr 21 Python
Python查看微信撤回消息代码
Jun 07 Python
Python实现读取SQLServer数据并插入到MongoDB数据库的方法示例
Jun 09 Python
python实现批量解析邮件并下载附件
Jun 19 Python
使用Flask-Cache缓存实现给Flask提速的方法详解
Jun 11 Python
python PIL和CV对 图片的读取,显示,裁剪,保存实现方法
Aug 07 Python
python框架Django实战商城项目之工程搭建过程图文详解
Mar 09 Python
python实现人性化显示金额数字实例详解
Sep 25 Python
python和C++共享内存传输图像的示例
Oct 27 Python
利用python为PostgreSQL的表自动添加分区
Jan 18 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
虫族 Zerg 历史背景
2020/03/14 星际争霸
初次接触php抽象工厂模式(Elgg)
2010/03/21 PHP
php UBB 解析实现代码
2011/11/27 PHP
LNMP部署laravel以及xhprof安装使用教程
2017/09/14 PHP
Ubuntu 16.04中Laravel5.4升级到5.6的步骤
2018/12/07 PHP
javascript删除option选项的多种方法总结
2013/11/22 Javascript
jquery实现手风琴效果
2015/11/20 Javascript
基于Jquery实现仿百度百科右侧导航代码附源码下载
2015/11/27 Javascript
深入解析AngularJS框架中$scope的作用与生命周期
2016/03/05 Javascript
基于bootstrap插件实现autocomplete自动完成表单
2016/05/07 Javascript
当jquery ajax遇上401请求的解决方法
2016/05/19 Javascript
jQuery实现鼠标经过时高亮,同时其他同级元素变暗的效果
2016/09/18 Javascript
JavaScript实现的商品抢购倒计时功能示例
2017/04/17 Javascript
Node.JS文件系统解析实例详解
2017/05/15 Javascript
Angular 5.0 来了! 有这些大变化
2017/11/15 Javascript
详解vue.js之props传递参数
2017/12/12 Javascript
vue初尝试--项目结构(推荐)
2018/01/30 Javascript
Vue使用mixins实现压缩图片代码
2018/03/14 Javascript
p5.js入门教程之鼠标交互的示例
2018/03/16 Javascript
微信小程序之swiper轮播图中的图片自适应高度的方法
2018/04/23 Javascript
Python操作列表的常用方法分享
2014/02/13 Python
浅析Python中signal包的使用
2015/11/13 Python
Python完成毫秒级抢淘宝大单功能
2019/06/06 Python
Django-Model数据库操作(增删改查、连表结构)详解
2019/07/17 Python
pytorch索引查找 index_select的例子
2019/08/18 Python
python装饰器的特性原理详解
2019/12/25 Python
Django之choices选项和富文本编辑器的使用详解
2020/04/01 Python
Python基于httpx模块实现发送请求
2020/07/07 Python
HTML5 Video标签的属性、方法和事件汇总介绍
2015/04/24 HTML / CSS
暇步士官网:Hush Puppies
2016/09/22 全球购物
普通PHP程序员笔试题
2016/01/01 面试题
应届毕业生专业个人求职自荐信格式
2013/11/20 职场文书
高中美术教师事迹材料
2014/08/22 职场文书
个人批评与自我批评
2014/10/15 职场文书
大学生入党自荐书
2015/03/05 职场文书
计算机教师工作总结
2015/08/13 职场文书