Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)


Posted in Python onFebruary 17, 2020

-----最近从github上找了一个代码跑,但是cpu训练的时间实在是太长,所以想用gpu训练一下,经过了一天的折腾终于可以用gpu进行训练了,嘿嘿~

首先先看一下自己电脑的显卡信息:

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)

可以看到我的显卡为MX250

然后进入NVIDIA控制面板->系统信息->组件 查看可以使用的cuda版本

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)

这里我先下载了cuda10.1的版本,不过后来我发现tensorflow-gpu 1.8.0仅支持cuda9.0的版本,所以之后我又重装了一遍cuda9.0,中间还经历了删除cuda10.0,两个版本的安装都是一样的。

进入官网:https://developer.nvidia.com/cuda-toolkit-archive 找到和显卡信息相匹配的cuda(cuda是向下兼容的)

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)

下载完成后按照默认的执行下去(当然也可以修改安装的路径)在安装时如果电脑装有vs2017,那么这里建议取消VS

 Integration

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)

然后一步步往下执行

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)

检查cuda是否安装成功:输入nvcc --version

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn) 

cuda的环境变量在安装时自动配置了,所以不需要我们操心

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)

下面安装对应版本的cudnn:https://developer.nvidia.com/cudnn 在这个网站先创建账户,再下载对应版本的cudnn

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn) 

cudnn下载完成后进行解压,里面有三个文件夹

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)

将上面的三个文件夹覆盖你之前安装的cuda路径下的相同名称的文件夹(默认安装的cuda和我截图中的路径一样)

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)

下面我们安装tensorflow-gpu = 1.8.0

因为我安装的版本比较老了,现在最新的Anaconda已经装不了了,这里把我的Anaconda版本是Anaconda3-5.1.0-Windows-x86_64,使用的python版本是3.6 安装的时候记得把环境变量勾选上,这样就不用自己配置了

安装完成后创建一个名称为tensorflow的环境: conda create --name tensorflow python=3.6

然后进入tensorflow环境中正式安装tensorflow-gpu

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)

然后输入命令:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.8.0

等待安装完成即可

下面进行tensorflow-gpu的测试

创建一个python文件,代码如下:

import tensorflow as tf
with tf.device('/cpu:0'):
 a = tf.constant([1.0,2.0,3.0],shape=[3],name='a')
 b = tf.constant([1.0,2.0,3.0],shape=[3],name='b')
with tf.device('/gpu:0'):
 c = a+b
sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True,log_device_placement=True))
sess.run(tf.global_variables_initializer())
print(sess.run(c))

在刚才激活的tensorflow环境下进行运行:python + 文件名

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)

结果如下:

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)

-----然后我运行了一下之前用cpu训练的代码,gpu训练的速度就是快,原先cpu要训练七八个小时的代码,gpu一个小时不到跑完了,还是爽啊,虽然我知道我的显卡很垃圾,,,,

Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程分析(显卡MX250+CUDA9.0+cudnn)

可以看到显卡使用的情况了,哈哈哈~~

总结

以上所述是小编给大家介绍的Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程(显卡MX250+CUDA9.0+cudnn),希望对大家有所帮助!

Python 相关文章推荐
Python Deque 模块使用详解
Jul 04 Python
Python语言的12个基础知识点小结
Jul 10 Python
Python中type的构造函数参数含义说明
Jun 21 Python
Python检测网站链接是否已存在
Apr 07 Python
python实现可逆简单的加密算法
Mar 22 Python
详解Python 切片语法
Jun 10 Python
Python打开文件、文件读写操作、with方式、文件常用函数实例分析
Jan 07 Python
Pytorch 高效使用GPU的操作
Jun 27 Python
从零开始的TensorFlow+VScode开发环境搭建的步骤(图文)
Aug 31 Python
深入浅析pycharm中 Make available to all projects的含义
Sep 15 Python
selenium+python实现基本自动化测试的示例代码
Jan 27 Python
pytorch中Schedule与warmup_steps的用法说明
May 24 Python
Windows下实现将Pascal VOC转化为TFRecords
Feb 17 #Python
tensorflow生成多个tfrecord文件实例
Feb 17 #Python
tensorflow将图片保存为tfrecord和tfrecord的读取方式
Feb 17 #Python
Python 读取有公式cell的结果内容实例方法
Feb 17 #Python
Python破解BiliBili滑块验证码的思路详解(完美避开人机识别)
Feb 17 #Python
Tensorflow 实现将图像与标签数据转化为tfRecord文件
Feb 17 #Python
将自己的数据集制作成TFRecord格式教程
Feb 17 #Python
You might like
PHP使用文件锁解决高并发问题示例
2018/03/29 PHP
jQuery 事件队列调整方法
2009/09/18 Javascript
基于jQuery的简单的列表导航菜单
2011/03/02 Javascript
JavaScript中的稀疏数组与密集数组[译]
2012/09/17 Javascript
将form表单中的元素转换成对象的方法适用表单提交
2014/05/02 Javascript
javascript中attribute和property的区别详解
2014/06/05 Javascript
javascript相关事件的几个概念
2015/05/21 Javascript
使用AngularJS处理单选框和复选框的简单方法
2015/06/19 Javascript
jQuery实现模仿微博下拉滚动条加载数据效果
2015/12/25 Javascript
json与jsonp知识小结(推荐)
2016/08/16 Javascript
JavaScript实现垂直向上无缝滚动特效代码
2016/11/23 Javascript
JavaScript 事件对内存和性能的影响
2017/01/22 Javascript
原生JS发送异步数据请求
2017/06/08 Javascript
代码详解Vuejs响应式原理
2017/12/20 Javascript
微信小程序收货地址API兼容低版本解决方法
2019/05/18 Javascript
在Layui 的表格模板中,实现layer父页面和子页面传值交互的方法
2019/09/10 Javascript
vue+Element中table表格实现可编辑(select下拉框)
2020/05/21 Javascript
一个检测OpenSSL心脏出血漏洞的Python脚本分享
2014/04/10 Python
Python ORM框架SQLAlchemy学习笔记之安装和简单查询实例
2014/06/10 Python
常见的python正则用法实例讲解
2016/06/21 Python
Pandas数据离散化原理及实例解析
2019/11/16 Python
完美解决Django2.0中models下的ForeignKey()问题
2020/05/19 Python
python实现在列表中查找某个元素的下标示例
2020/11/16 Python
HTML5的结构和语义(3):语义性的块级元素
2008/10/17 HTML / CSS
伦敦平价潮流珠宝首饰品牌:Astrid & Miyu
2016/10/10 全球购物
中职应届生会计求职信
2013/10/23 职场文书
机电职业生涯规划书范文
2014/03/08 职场文书
请假条标准格式规范
2014/04/10 职场文书
安全生产责任书范本
2014/04/15 职场文书
作文评语大全
2014/04/23 职场文书
竞聘演讲稿开场白
2014/08/25 职场文书
订货会邀请函
2015/01/31 职场文书
表扬信范文
2015/05/04 职场文书
Python基础之字符串格式化详解
2021/04/21 Python
Python 快速验证代理IP是否有效的方法实现
2021/07/15 Python
用JS写一个发布订阅模式
2021/11/07 Javascript