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中逻辑运算符的使用
May 13 Python
python获取当前用户的主目录路径方法(推荐)
Jan 12 Python
详解python里使用正则表达式的分组命名方式
Oct 24 Python
Python cookbook(数据结构与算法)找到最大或最小的N个元素实现方法示例
Feb 13 Python
Python实现Dijkstra算法
Oct 17 Python
详解分布式任务队列Celery使用说明
Nov 29 Python
python实现flappy bird小游戏
Dec 24 Python
python3 tcp的粘包现象和解决办法解析
Dec 09 Python
python opencv pytesseract 验证码识别的实现
Aug 28 Python
Python类的继承super相关原理解析
Oct 22 Python
python+selenium实现12306模拟登录的步骤
Jan 21 Python
Python pyecharts案例超市4年数据可视化分析
Aug 14 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中inlcude()性能对比详解
2012/09/16 PHP
php实现的click captcha点击验证码类实例
2014/09/23 PHP
php用ini_get获取php.ini里变量值的方法
2015/03/04 PHP
Laravel 5.3 学习笔记之 配置
2016/08/28 PHP
Ext JS 4实现带week(星期)的日期选择控件(实战一)
2013/08/21 Javascript
jQuery+ajax实现无刷新级联菜单示例
2015/05/21 Javascript
Angular2表单自定义验证器的实现
2016/10/19 Javascript
JS日期对象简单操作(获取当前年份、星期、时间)
2016/10/26 Javascript
JavaScript函数节流的两种写法
2017/04/07 Javascript
three.js实现3D视野缩放效果
2017/11/16 Javascript
javascript中的隐式调用
2018/02/10 Javascript
JavaScript代码实现txt文件的上传预览功能
2018/03/27 Javascript
vue-cli构建项目下使用微信分享功能
2018/05/28 Javascript
js继承的这6种方式!(上)
2019/04/23 Javascript
jQuery冲突问题解决方法
2021/01/19 jQuery
[05:08]DOTA2-DPC中国联赛3月6日Recap集锦
2021/03/11 DOTA
使用Python进行新浪微博的mid和url互相转换实例(10进制和62进制互算)
2014/04/25 Python
Python编程实现生成特定范围内不重复多个随机数的2种方法
2017/04/14 Python
用TensorFlow实现多类支持向量机的示例代码
2018/04/28 Python
python读取excel指定列数据并写入到新的excel方法
2018/07/10 Python
使用 Python 实现微信群友统计器的思路详解
2018/09/26 Python
Python enumerate函数功能与用法示例
2019/03/01 Python
python语言基本语句用法总结
2019/06/11 Python
应用OpenCV和Python进行SIFT算法的实现详解
2019/08/21 Python
tensorflow的计算图总结
2020/01/12 Python
Jupyter加载文件的实现方法
2020/04/14 Python
python如何修改文件时间属性
2021/02/05 Python
英国男士时尚网站:Dandy Fellow
2018/02/09 全球购物
具有防紫外线功能的高性能钓鱼服装:Hook&Tackle
2018/08/16 全球购物
乡镇务虚会发言材料
2014/10/20 职场文书
2015新年寄语(一句话)
2014/12/08 职场文书
大专护理专业自荐信
2015/03/25 职场文书
办公室日常管理制度
2015/08/04 职场文书
预备党员入党感想
2015/08/10 职场文书
纪律委员竞选稿
2015/11/19 职场文书
《角的初步认识》教学反思
2016/02/17 职场文书