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 条评论
登录 后参与评论

相关文章

来自专栏about云

spark开发环境详细教程2:window下sbt库的设置

问题导读 1.sbt在IntelliJ IDEA里面比较慢,该如何解决? 2.如何在window里面更改阿里源? 3.如何在Linux里更改源? 上一篇...

3006
来自专栏一个默默无闻的工程师的日常

rabbitmq无法重新加入集群,启动失败的问题

1171
来自专栏新工科课程建设探讨——以能源与动力工程专业为例

2.1 开发平台搭建: 工欲善其事,必先利其器

440
来自专栏逸鹏说道

C#通过WMI的wind32 的API函数实现msinfo32的本地和远程计算机的系统摘要信息查看功能

最近做一个项目碰到要实现查看本地和远程计算机的摘要信息,采用命令行msinfo32可以很快查看到,如下图: ? 需要在用C#来实现类似信息查看。尤其远程计算机的...

3365
来自专栏Linyb极客之路

Maven 私服配置

注意:这样的配置(第一个仓库是代码库,第二个是插件库)只对当前的Maven项目有效。实际的项目开发中,我们都想着一次配置就能让本机的所有Maven项目都使用Ma...

712
来自专栏月色的自留地

三种方式给apt设置代理

有很多第三方工具可以用,比如proxychains,非常好用,不过今天这不是正题。因为有可能没有代理,上网你都做不到,更别提下载软件了。想一想方法还是告诉你,免...

501
来自专栏AI科技大本营的专栏

TensorFlow集成TensorRT环境配置

本文前提是cuda和cudnn以及TensorRT已经安装完毕,具体详情,可以参考上一篇文章: https://developer.nvidia-china.c...

4785
来自专栏MoeLove

Git 本地仓库和裸仓库

通常我们会用 git init 命令来将我们所在的目录转换为一个 Git 本地仓库或者初始化一个新的空仓库。

732
来自专栏C/C++基础

Linux命令(5)——shutdown命令

使用shutdown命令时,在系统关机前,可以通知所有登录者系统将要关闭。此时login指令会被冻结,新用户将不能再登录,这是推荐使用的安全关机方式。关机之前,...

764
来自专栏云计算教程系列

如何在CentOS 7上安装和配置Zabbix以安全地监视远程服务器

Zabbix是用于网络和应用程序的开源监控软件。它提供对从服务器,虚拟机和任何其他类型的网络设备收集的数千个指标的实时监控。这些指标可帮助您确定IT基础架构当前...

250

扫码关注云+社区