Window TensorFlow(gpu)安装及坑的解决
来源:互联网 发布:站长工具源码 编辑:程序博客网 时间:2024/06/05 19:32
1、window下TensorFlow版本支持说明
从上面的说明知道现在TensorFlow在window下只支持 Python3.5,所以在安装TensorFlow时,Python环境一定选择Python3.5(比如你选择通过Anaconda来安装Python环境话一定要选Python3.5 Anaconda4.2.0)。
2、安装TensorFlow(GPU)
(1)Anaconda下载安装:Window下安装非常简单,直接运行安装包下一步下一步就行,安装过程会有选择打钩的地方,第一个是添加环境变量,第二个是把Anaconda下的Python设置为默认版本,这里建议都勾选。
(2)首先确保你的电脑安装了Visual Studio2010以上版本,因为在安装cuda时需要编译环境。
(3)安装cuda,我安装的是cuda8.0,下载地址点击打开链接
根据自己的环境选择对应的版本,exe分为网络版和本地版,网络版安装包比较小,执行安装的时候再去下载需要的包;本地版安装包是直接下载完整安装包。这里建议下载本地版(网络版我试过,安装一晚上失败了…),另外建议用win10自带的edge浏览器下载,速度比谷歌啥的快多了。
下载下来后就正常安装就可以了。 验证cuda8.0已正确安装:打开cmd,输入$ nvcc -V,结果如下图:
(4)按照TensorFlow官网教程直接安装TensorFlow(GPU)版本即可。
(5)下载cudnn,cudnn的版本要和cuda8.0版本对应,不然会出现错误。下载地址。
3、一些坑的解决
当我安装完成准备开一波车的时候,出来各种错,作为年轻的老司机我讲讲经验。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对应的文件夹下面即可。
(2).example 测试
下载github上的tensorflow源码,里面有很多示例代码
run example:
python mnist_with_summaries.py..............................
结果刚发车就熄火了…couldn't open CUDA library cupti64_80.dll
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\dso_loader.cc:119] Couldn't open CUDA library cupti64_80.dll
查了一下,这个dll在NVIDIA GPU Computing Toolkit\CUDA\v8.0\extras\CUPTI\libx64
,因为这个也没添加到系统变量,简单的处理方式,把这里面的文件也拷到环境变量的bin文件夹下。
再次启动,终于愉快的run起来了。
(3) check failed : s.ok() could not find cudnncreate in cudnn DSO 问题
出现该问题就是因为你选的cuda版本和cudnn版本不对应,重新安装cudnn即可。
- Window TensorFlow(gpu)安装及坑的解决
- tensorflow -GPU 安装的坑
- tensorflow及GPU加速在win10下的安装教程
- win7安装tensorflow(GPU版)及环境变量设置
- ubuntu16.04下安装CUDA cuDNN及tensorflow-gpu版本及caffe-gpu过程(初版)
- tensorflow(GPU版)安装
- CPU和GPU的区别、工作原理、及如何tensorflow-GPU安装等操作
- windows10安装tensorflow-GPU及环境配置
- 简单安装tensorflow-gpu(GPU版tensorflow)
- TensorFlow 及 Keras 在window上的安装
- win10下安装tensorflow-gpu遇到的坑
- windows 下 TensorFlow(GPU 版)的安装
- Tensorflow + Caffe + Torch 的详细安装指南(GPU版本)
- (转)tensorflow中使用指定的GPU及GPU显存
- tensorflow中使用指定的GPU及GPU显存
- Win10下安装GPU版本的tensorflow
- Anaconda Tensorflow GPU 版本的安装问题
- ubuntu16.04安装gpu版本的tensorflow
- 机器学习之期望最大算法(EM算法)
- 成功搭建 Jenkins+ant+jmeter 进行接口自动化测试分享
- python3 tkinter如何绑定事件
- unit15 系统虚拟机管理
- MySQL CRASH COURSE命令总结(第四章检索数据)
- Window TensorFlow(gpu)安装及坑的解决
- 定时任务执行两次
- Struts2:strut2的工作原理
- tsung转载整理 笔记1
- 关于反编译后显示盗版
- node+express 项目中ejs模板的使用
- C++11 封装小觑:从 std::array 看 C++11 风格
- php的延迟静态绑定
- 详解Paint的setXfermode(Xfermode xfermode)