win10下配置Tensorflow1.4

今天碰巧帮人配了一台win10 + CUDA8.0 + CUDNN V6.1 + tf 1.4,特此记录

配置要求

window版本的tf是用不成python 2.X,至少是python 3.5,看官方要求吧,https://www.tensorflow.org/install/install_windows,如下所示:

所以总结下就是:

  1. Python 3.5 or later
  2. CUDA 8.0 or later
  3. CUDNN V6.1 or later

具体CUDA和CUDNN的版本要求取决于你要装的tf的版本要求,1.4至少要CUDA 8.0 + CUDNN V6.1,故此我安装的配置就是: python 3.6 + CUDA 8.0 + CUDNN V6.1 + tf 1.4

接下来就一个一个装吧

python安装

先下载python 3.6,https://www.python.org/downloads/release/python-362/

选择对应的平台,下载下来是个exe,直接双击安装就好了,记得勾选把python添加的环境变量PATH中这个选项,不然你后续还要自己配下

CUDA

可以官网选择下载,最新版是CUDA 9,https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10

有local(约1.2G)和network可选,区别就是network安装时需要联网,建议还是local

上面是CUDA 9.0的,8.0的版本也甩个链接吧: https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda_8.0.44_win10-exe local的哈,大约1.3G,自己看着流量~~

下载下来,双击就可以开始安装了,照着屏幕提示就可以了

CUDNN

首先还是下载: step 1: 上官网 https://developer.nvidia.com/cudnn,去注册下,然后登陆才能下载 step 2: 先从最新的V7开始点击download,后面可以选择版本

然后按照需要的进行选择就可以了

下载下来解压,内部目录是这样的:

就是3个目录bin、include、lib,全部拷贝到你CUDA的目录下就行了,CUDA的目录默认在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,直接ctrl+c,ctrl+v粘贴过去就好了

tf的安装

首先要安装必要的python库,有点多,可以用pip安装,但是我建议直接安装anacoda,它集成了这些必要的库 下载安装: https://www.anaconda.com/download/ 安装完以后,诸如numpy scipy等其实都已经安装好了,接下来安装tf tf官网速度很慢,推荐使用国内的清华源,速度快

[修改源为清华源] conda config --add channels 'https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/'

[设置搜索时显示通道地址] conda config --set show_channel_urls yes

配置完以后,文件C:\Users\USER_NAME\.condarc就是配置文件,可以手动编辑或者删除

[搜索tf] anaconda search -t conda tensorflow-gpu 接下来选择一个源地址,假设你选的是xxx

[安装tf] anaconda show XXX/tensorflow-gpu 最后一行会出现安装方式,大概是这样的: conda install --channel https://conda.anaconda.org/XXX tensorflow-gpu

OK了,等着装吧,速度应该还可以

测试

安装好了就测试下吧,先试试能不能导入tf:

import tensorflow as tf

如果没有错就应该是安装好了,否则自己根据错误提示去debug吧

再试试能不能做运算:

sess = tf.Session()
a = tf.constant(2)
b = tf.constant(3)
print(sess.run(a+b))

要是这步你能看见自己的GPU信息,比如1070 8GB xxx等信息,就应该是没问题了

到这都没问题的话,就已经OK了!

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏mathor

网络协议详解

地址解析协议,即ARP(Address Resolution Protocol),是根据IP地址获取物理地址的一个TCP/IP协议。主机发送消息时将包含目标I...

903
来自专栏我的博客

CSRF原理和防范

一.CSRF CSRF(Cross-site request forgery),中文名称:跨站请求伪造,也被称为:one click attack/sessi...

3758
来自专栏专注 Java 基础分享

看懂「www.google.com」背后的逻辑

在前两篇文章中,我们完整的描述了计算机网络 OSI 五层模型的相关内容。那么,本篇将会从一个实践案例开始,带你从整体上重新认识我们的计算机网络。

850
来自专栏windealli

socket常用函数知识点整理

调用close() 之后,进程不能再使用该描述符。 但是已经发送队列中的数据还是会继续发送,等到发送回再发起四次挥手。

2902
来自专栏Java架构师历程

TCP连接的状态详解以及故障排查

我们通过了解TCP各个状态,可以排除和定位网络或系统故障时大有帮助。(总结网络上的内容)

4852
来自专栏运维技术迷

win7无法保存打印机设置(0x000006d9)解决方法

今天在给同事共享打印机的时候一直提示这个错误,因为我们是域环境,当时还以为是域权限的设置问题,但看了看域策略也没对这一部分进行权限设置呀!索性就度娘一下,找了一...

2964
来自专栏北京马哥教育

TCP恋爱史:三次握手和四次分手

TCP恋爱史:三次握手和四次分手 ---- 一个应用占用CPU很高,除了确实是计算密集型应用之外,通常原因都是出现了死循环。 以我们最近出现的 TCP协议非常重...

2483
来自专栏网络 后台

TCP TIME_WAIT

从图中可以看出,若服务器主动关闭连接,在四次挥手的最后一个ACK后连接端口会变为TIME_WAIT状态, 状态停留时长为两个MSL(最大分段寿命),这个状态只有...

1480
来自专栏用户2442861的专栏

TCP/IP详解之 《网络协议》图解 TCP 连接建立与释放

http://blog.csdn.net/chenhanzhun/article/details/41622555

1321
来自专栏向治洪

maven配置详解

什么是pom?     pom作为项目对象模型。通过xml表示maven项目,使用pom.xml来实现。主要描述了项目:包括配置文件;开发者需要遵循的规则,缺...

2035

扫码关注云+社区