win10+cuda8.0+cudnn+Tensorflow(GPU)安装
来源:互联网 发布:数据库原理及应用答案 编辑:程序博客网 时间:2024/05/21 22:56
传说中的Tensorflow终于支持windows了,下面介绍一下Win10下Tensorflow的安装与使用
准备工作:
1.python3.5(64位)目前Tensorflow只支持64位python3.5以上版本
下载链接:
https://www.python.org/ftp/python/3.5.3/python-3.5.3rc1-amd64.exe
2.安装numpy
python3.5以上自带pip模块,打开终端输入pip3 install numpy
3.cuda+cudnn
这部分参考的:
http://blog.csdn.net/sb19931201/article/details/53648615
(1).cuda v8.0安装包下载 ,还有安装指南之类的,稍微看下就行。
根据自己的环境选择对应的版本,exe分为网络版和本地版,网络版安装包比较小,执行安装的时候再去下载需要的包;本地版安装包是直接下载完整安装包。这里建议下载本地版(网络版我试过,安装一晚上失败了…),另外建议用win10自带的edge浏览器下载,速度比谷歌啥的快多了。
下载下来后就正常安装就可以了。
(2).cuDnn库下载
下载这个安装包需要注册并且填一堆问卷,下下来以后把相关包不用安装,直接拷到cuda路径对应的文件夹下面就行(后面说)。
(3).cuda测试
在cuda安装好后会自动下载一些example,自己运行测试一下,问题不大。
(4).关键坑(运行TensorFlow)
当我安装完成准备开一波车的时候,出来各种错,作为年轻的老司机我讲讲经验。TensorFlow 是一个编程系统, 使用图来表示计算任务,图必须在Session(会话)里被启动. Session将图的op(操作)分发到诸如CPU或GPU之类的设备上运行。所以,这个时候你运行python然后import tensorflow as tf
是不会报错的,但是当你要执行tf.Session()
的时候可能就有问题了。这个时候将会调用cuda,我在这里遇到的问题是各种lib,dll加载不了。经过一番检查,定位到问题,cuda安装完成后默认的环境变量配置不对,CUDA_PATH是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
,但是这样不能直接访问到bin和lib\x64下的程序包,在path中加上这两个路径即可。
再发一波车,结果还有问题…还有一个库加载不了,就是上面的cuDnn库了,很简单,解压刚才下下来的安装包,将这三个文件夹下的文件拷到CUDA对应的文件夹下面即可。
这样就大功告成了。
最关键一步:
pip3 install tensorflow-gpu
哈哈哈,就这么简单,快去测试一下吧
#coding=utf-8
import tensorflow as tf
import numpy as np
hello=tf.constant('hhh')
sess=tf.Session()
print (sess.run(hello))
成功的样子:
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\dso_loader.cc:128] successfully opened CUDA library cublas64_80.dll locally
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\dso_loader.cc:128] successfully opened CUDA library cudnn64_5.dll locally
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\dso_loader.cc:128] successfully opened CUDA library cufft64_80.dll locally
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\dso_loader.cc:128] successfully opened CUDA library nvcuda.dll locally
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\dso_loader.cc:128] successfully opened CUDA library curand64_80.dll locally
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\core\common_runtime\gpu\gpu_device.cc:885] Found device 0 with properties:
name: GeForce GTX 1060
major: 6 minor: 1 memoryClockRate (GHz) 1.6705
pciBusID 0000:01:00.0
Total memory: 6.00GiB
Free memory: 5.01GiB
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\core\common_runtime\gpu\gpu_device.cc:906] DMA: 0
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\core\common_runtime\gpu\gpu_device.cc:916] 0: Y
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\core\common_runtime\gpu\gpu_device.cc:975] Creating TensorFlow device (/gpu:0) -> (device: 0, name: GeForce GTX 1060, pci bus id: 0000:01:00.0)
b'hhh'
阅读全文
0 0
- win10+cuda8.0+cudnn+Tensorflow(GPU)安装
- win10+cuda8.0+cudnn+Tensorflow(GPU)安装
- Win10下安装Tensorflow(GPU)+CUDA8.0+cudnn6
- Win10下安装Tensorflow(GPU)+CUDA8.0+cudnn6
- Win10下安装Tensorflow(GPU)+CUDA8.0+cudnn6
- win10+anaconda2+cuda8.0+cudnn6.0安装tensorflow-gpu
- ubuntu14.04下CUDA8.0+cuDNN+tensorflow(with gpu support)安装教程
- Ubuntu 16.04安装cuda8.0 + cudnn 5.1 + tensorflow-gpu的正确姿势
- ubuntu16.04 +NVIDIA驱动+cuda8.0+cudnn+andaconda+tensorflow(GPU版)+Spyder+pycharm全套配置
- Ubuntu 14.04 安装 CUDA8.0 cudnn 5.1 tensorflow1.2.1GPU
- Ubuntu16.04安装caffe+CUDA8.0+cuDNN GPU version
- Ubuntu17.04 CUDA8.0 Cudnn v7 tensorflow1.3-GPU安装
- 0-0不走坑路, 手把手教你Linux16.04, Anaconda3, CUDA8.0, cudnn v5.1 tensorflow GPU版, opencv3安装
- ubantu16.04安装tensorflow(GPU)+cuda8.0+cudnn6.0
- win10下Visual Studio2015+cuda8.0+cudnn v5.1+Anaconda4.4.0+tensorflow+Keras+Pycharm+OpenCV3.2.0安装教程
- 在win10下安装Tensorflow(CUDA8.0+cuDNN5.1)
- Ubuntu14.04 & CUDA8.0 & Theano & Tensorflow & TensorLayer & Cudnn安装血泪史
- windows10+cuda8.0+cudnn+python3.5+tensorflow-gpu-1.3.0-rc0( bottleneck() argument after ** must be )
- 数据传递之startActivityForResult
- 本地windows跑Python程序调用Spark
- android llvm-o混淆
- 数据结构--树的基本概念
- TensorFlow的reshape操作 tf.reshape
- win10+cuda8.0+cudnn+Tensorflow(GPU)安装
- 利用插件保存unity3D切割过的sprite文件
- 5G 网络构架解读
- Sqlserver2008及以上使用全文索引排除干扰词
- 【Scikit-Learn 中文文档】内核岭回归
- JAVA 第7次作业
- mips架构linux启动分析(五)(bootmem和blockmem机制的初始化)(prom_init_numa_memory)
- 11月20日云栖精选夜读:围观阿里总部:边喝茶边搞技术是一种怎样的体验?
- Java基础——Java NIO详解(一)