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即可。


0 0
原创粉丝点击