博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
win10 + cuda(v9.0) 安装TensorFlow-gpu版
阅读量:7298 次
发布时间:2019-06-30

本文共 1155 字,大约阅读时间需要 3 分钟。

之前在实习公司的电脑上装过TensorFlow-gpu,那时候很快就装好了。但在自己的笔记本上装时,却搞了很久。。。

一部分原因是因为用校园网下载cuda toolkit 和cudnn ,总是在最后时刻罢工。辗转叫了几个人帮忙,才把这两个东西下载好。。

安装步骤

其实,这个安装过程还是蛮简单的。据说cuda版本和cudnn版本没选好的话,就会遇到很多坑,所幸,我选的版本都能很好的驾驭tensorflow。

下面是要安装的东西,可以依次安装,也可以按你喜欢的顺序来安装。但要保证安装了VC++插件和python之后,才安装tensorflow。

  • Visual C++ Redistributable for Visual Studio 2015: tensorflow其实是基于VC++2015开发的,所以需要安装。下载后,直接默认安装就行

  • python环境:我用的是anaconda3.x,新建了一个python3.5的环境,然后在python3.5中安装的tensorflow-gpu。这个不会的话,可以尝试搜索“anaconda安装、anaconda清华源安装”和“anaconda新建python环境”。。

  • cuda9.0: .下载之后,默认安装就行,中间要选同意的地方就选同意。

这里写图片描述

  • 安装好cuda后,要配置系统环境。用小娜搜索“控制面板”–>选择大图标–>选择系统–>选择高级系统设置–>选择环境变量–>选择path,新增加“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0”–>确定。。然后再新增几个变量,如下图(已经存在的就不用增加了):

    这里写图片描述

  • cudnn:() 下载好后,解压,把所有东西复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 中,就ok啦

  • tensorflow-gpu:安装win+R,运行cmd,打开命令窗口,激活python3.5的环境(activate 环境名),然后输入pip install tensorflow-gpu 即可。等待安装成功就行。

测试

我喜欢用jupyter notebook来写python,所以现在也用它来写一下测试代码:

import tensorflow as tfhi = tf.constant('Hi , welcome to TensorFlow ! ')sess = tf.Session()print(sess.run(hi).decode())

得到结果如下:

这里写图片描述

cool,顺利输出了结果,说明安装成功。

这里写图片描述

转载于:https://www.cnblogs.com/surecheun/p/9648949.html

你可能感兴趣的文章
java 调用js
查看>>
iOS开发UI篇—Quartz2D使用(图形上下文栈)
查看>>
Oracle迁移MySQL笔记
查看>>
Building a Pub/Sub Message Bus with Wcf,Msmq,IIS
查看>>
Mybatis实现批量删除
查看>>
【leetcode】995. Minimum Number of K Consecutive Bit Flips
查看>>
【洛谷 P4886】 快递员 (点分治)
查看>>
在Ajax中将数组转换成字符串(0517-am)
查看>>
hive字符串函数
查看>>
【erlang ~ 4 days】 Day # 1.2 Sequential Programming
查看>>
HDFS Erasure Coding介绍
查看>>
abstract vs interface
查看>>
egret 游戏优化文档
查看>>
蚂蚁金服研发面经
查看>>
xmanagr 注册机执行ubuntu 桌面程序,ubuntu无需安装 桌面环境
查看>>
开源存储
查看>>
sqlplus乱码
查看>>
CodeForces 213C Relay Race :从左上角到右下角再返回,每个价值计算一次,问最多收集价值 :dp...
查看>>
EFCore中SQLSERVER 2008 的分页问题
查看>>
Python中变量的绑定,或者说引用
查看>>