windows安装tensorflow-gpu开发环境
新版的tensorflow已经支持windows了,不过python版本不能用2.x, 要用3.x,好像最新的tensorflow支持3.5和3.6这两个python版本,其他3.x的python版本没有测试过。当然电脑要有nvidia的显卡才行,我的电脑比较差,带的是gtx 950,跑起来例子比cpu还是快不少。下面是具体的安装步骤:
1. 安装最新版的tensorflow-gpu
pip install tf-nightly-gpu
注意:这个可能会比较慢,最好配置一下国内的pip源,我用的是清华大学的pip源,具体配置参考:
https://mirrors.tuna.tsinghua.edu.cn/help/pypi/
2. 安装cuda+cudnn
这里特别要注意,cudnn和cuda的版本要对应起来,不能乱装。我这里装的cuda9.0+cudnn 7 for duda 9.0
还有就是tensorflow目前这个版本:tf-nightly-gpu (1.6.0.dev20180126) 不支持cuda 9.1,我就是先装了9.1才发现它只支持cuda9.0,然后又卸载重装cuda 9.0。
2.1 cuda安装
https://developer.nvidia.com/cuda-90-download-archive
到这个页面选择相应系统的安装包,下载下来安装就ok了。
安装就装qq一样,傻瓜的点击下一步就行了。
2.2 cudnn 7 for cuda 9.0 安装
这个比较坑,为什么说比较坑呢,因为cudnn必须要和cuda的版本对应起来才行,否则出现问题就很麻烦,我这里下的cuda是9.0,所以cudnn要选择:cuDNN v7.0.5 (Dec 5, 2017), for CUDA 9.0
官方的下载地址(需要注册nvidia才能下载):
https://developer.nvidia.com/rdp/cudnn-download
其实根本就不用安装,把下载好的zip文件解压,然后把里面的文件拷贝到cuda安装位置的目录下面就行了,具体有三个文件要拷贝到cuda的三个目录下。
这些弄好以后,就完成了,可以打开cmd,输入python,进入python的shell中:
上一篇: DataVec/转换-2
下一篇: spring-boot开发一堆问题
推荐阅读
-
在Windows XP下安装Apache+MySQL+PHP环境
-
linux下eclipse+pdt(PHP集成开发环境安装)
-
Windows下安装PHP单元测试环境PHPUnit图文教程_php技巧
-
windows 7 搭建PHP 开发环境
-
Windows下筹建PHP开发环境
-
Windows环境下安装MySQL
-
Windows Server2008 R2 MVC 环境安装配置教程
-
Windows下安装PHP单元测试环境PHPUnit图文教程_PHP
-
windows下配置php5.5开发环境及开发扩展,windowsphp5.5
-
Linux服务器开发环境搭建 Nginx+PHP+MongoDB linux 连接mongodb mongodb安装教程 mongod