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每次处理固定个数的字符的方法总结
Jan 29 Python
在Python的Django框架上部署ORM库的教程
Apr 20 Python
Python访问纯真IP数据库脚本分享
Jun 29 Python
python根据日期返回星期几的方法
Jul 06 Python
用ReactJS和Python的Flask框架编写留言板的代码示例
Dec 19 Python
Python3安装Pillow与PIL的方法
Apr 03 Python
连接pandas以及数组转pandas的方法
Jun 28 Python
python应用文件读取与登录注册功能
Sep 23 Python
python实现图片插入文字
Nov 26 Python
在 Pycharm 安装使用black的方法详解
Apr 02 Python
Python中使用threading.Event协调线程的运行详解
May 02 Python
用gpu训练好的神经网络,用tensorflow-cpu跑出错的原因及解决方案
Mar 03 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制作静态网站的模板框架(二)
2006/10/09 PHP
phpmyadmin配置文件现在需要绝密的短密码(blowfish_secret)的2种解决方法
2014/05/07 PHP
php抓取网站图片并保存的实现方法
2015/10/29 PHP
用JS实现一个TreeMenu效果分享
2011/08/28 Javascript
contains和compareDocumentPosition 方法来确定是否HTML节点间的关系
2011/09/13 Javascript
js性能优化 如何更快速加载你的JavaScript页面
2012/03/17 Javascript
jQuery 鼠标经过(hover)事件的延时处理示例
2014/04/14 Javascript
jquery easyui使用心得
2014/07/07 Javascript
JS实现的N多简单无缝滚动代码(包含图文效果)
2015/11/06 Javascript
js点击按钮实现带遮罩层的弹出视频效果
2015/12/19 Javascript
实例讲解JavaScript中instanceof运算符的用法
2016/06/08 Javascript
BootStrap Fileinput的使用教程
2016/12/30 Javascript
jQuery Form表单取值的方法
2017/01/11 Javascript
vue移动端UI框架实现QQ侧边菜单组件
2018/03/09 Javascript
Layer组件多个iframe弹出层打开与关闭及参数传递的方法
2019/09/25 Javascript
JQuery常用选择器功能与用法实例分析
2019/12/23 jQuery
Vue+webpack实现懒加载过程解析
2020/02/17 Javascript
11个Python3字典内置方法大全与示例汇总
2019/05/13 Python
python实现对象列表根据某个属性排序的方法详解
2019/06/11 Python
Python模块/包/库安装的六种方法及区别
2020/02/24 Python
python标准库OS模块详解
2020/03/10 Python
python爬虫容易学吗
2020/06/02 Python
手摸手教你用canvas实现给图片添加平铺水印的实现
2019/08/20 HTML / CSS
Crabtree & Evelyn英国官网:瑰珀翠护手霜、香水、沐浴和身体护理
2018/04/26 全球购物
加拿大床上用品、家居装饰、厨房和浴室产品购物网站:Linen Chest
2018/06/05 全球购物
10条PHP编程习惯
2014/05/26 面试题
中专生学习生活的自我评价分享
2013/10/27 职场文书
初入社会应届生求职信
2013/11/18 职场文书
《陋室铭》教学反思
2014/02/26 职场文书
红领巾广播站广播稿
2014/10/19 职场文书
世界气象日活动总结
2015/02/27 职场文书
任命书标准格式
2015/03/02 职场文书
2015年社会治安综合治理工作总结
2015/04/10 职场文书
暗恋桃花源观后感
2015/06/12 职场文书
2019单位介绍信怎么写
2019/06/24 职场文书
pytorch 实现多个Dataloader同时训练
2021/05/29 Python