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实现获取网站PR及百度权重
Jan 21 Python
Python IDE PyCharm的基本快捷键和配置简介
Nov 04 Python
Python的pycurl包用法简介
Nov 13 Python
Windows下anaconda安装第三方包的方法小结(tensorflow、gensim为例)
Apr 05 Python
解决python大批量读写.doc文件的问题
May 08 Python
python破解zip加密文件的方法
May 31 Python
python 将大文件切分为多个小文件的实例
Jan 14 Python
三步实现Django Paginator分页的方法
Jun 11 Python
pygame实现非图片按钮效果
Oct 29 Python
python读写Excel表格的实例代码(简单实用)
Dec 19 Python
Python使用扩展库pywin32实现批量文档打印实例
Apr 09 Python
Django表单提交后实现获取相同name的不同value值
May 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
收音机史话 - 1960年代前后的DIY
2021/03/02 无线电
我的论坛源代码(八)
2006/10/09 PHP
浅析php中抽象类和接口的概念以及区别
2013/06/27 PHP
前端必学之PHP语法基础
2016/01/01 PHP
PHP共享内存使用与信号控制实例分析
2018/05/09 PHP
关于laravel-admin ueditor 集成并解决刷新的问题
2019/10/21 PHP
TP5框架实现签到功能的方法分析
2020/04/05 PHP
从阿里妈妈发现的几个不错的表单验证函数
2007/09/21 Javascript
JavaScript高级程序设计 学习笔记 js高级技巧
2011/09/20 Javascript
jquery实现弹出层完美居中效果
2014/03/03 Javascript
JQuery为页面Dom元素绑定事件及解除绑定方法
2014/04/23 Javascript
js实现方块上下左右移动效果
2017/08/17 Javascript
vue中appear的用法
2017/08/17 Javascript
jQuery实现动态显示select下拉列表数据的方法
2018/02/05 jQuery
微信小程序6位或多位验证码密码输入框功能的实现代码
2018/05/29 Javascript
jQuery实现下拉菜单动态添加数据点击滑出收起其他功能
2018/06/14 jQuery
使用pkg打包Node.js应用的方法步骤
2018/10/19 Javascript
详解vue 2.6 中 slot 的新用法
2019/07/09 Javascript
微信自定义分享链接信息(标题,图片和内容)实现过程详解
2019/09/04 Javascript
[14:24]Optic Gaming vs PSG LGD BO3
2018/06/07 DOTA
Python利用flask sqlalchemy实现分页效果
2020/08/02 Python
浅谈Python中带_的变量或函数命名
2017/12/04 Python
Python用 KNN 进行验证码识别的实现方法
2018/02/06 Python
根据DataFrame某一列的值来选择具体的某一行方法
2018/07/03 Python
django 实现编写控制登录和访问权限控制的中间件方法
2019/01/15 Python
Django 中自定义 Admin 样式与功能的实现方法
2019/07/04 Python
Python threading.local代码实例及原理解析
2020/03/16 Python
Python+Appium实现自动化清理微信僵尸好友的方法
2021/02/04 Python
日语系毕业生推荐信
2013/11/11 职场文书
2014道德模范事迹材料
2014/02/16 职场文书
优秀应届毕业生推荐信
2014/02/18 职场文书
运动会广播稿20字
2014/02/18 职场文书
环境科学专业求职信
2014/08/04 职场文书
2015年销售人员工作总结
2015/04/07 职场文书
对学校的意见和建议
2015/06/04 职场文书
使用Canvas绘制一个游戏人物属性图
2022/03/25 Javascript