Windows 7/8.1 下安装 GPU版本的 tensorflow

来源:互联网 发布:小米关闭免费网络短信 编辑:程序博客网 时间:2024/06/13 05:29

最近 人工智能大火,tensorflow作为Google提供的最有力的深度学习开源框架,不得不建议大家赶紧下手尝试。下面我们介绍下如何在 Windows下使用GPU版本的tensorflow,哦,不是使用 ,是安装。


首先,查看你的电脑是否支持独立的NVIDIA显卡,只要看看有没有贴NVIDIA绿色眼睛的标志就好。


确定之后,请先下载安装CUDA库,目前最新版本为CUDA 8.0,下载地址:点击打开链接



按照你的系统配置选择相应的版本。最后选择下载网络版的驱动安装引导工具。点击安装,都默认,它会自动帮你配置好。


顺带,我们需要下载基于CUDA 的GPU加速深度神经网络(DNN)算法库,cudnn.

注册一个NVIDIA 账号:注册地址:点击打开链接

注册号=好之后,这里下载:点击打开链接


下载后解压。在安装CUDA的位置将解压好的CuDNN的三个文件夹 lib,bin,include 拷贝到CUDA 安装路径下。不知道CUDA安装路径的

同学请查看自己系统的环境变量设置。右击 我的电脑-> 属性->高级系统设置->环境变量->找到CUDA_v8的默认路径。


双击,拷贝路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0(这是我的电脑)。将刚才解压好的 cuDNN下的三个文件全部拷贝到这个路径下对应的目录下,只需要粘贴,选择替换就好。



这些工作OK之后,安装Python3.5。目前tensorflow还不支持Windows下高于3.5版本的Python。对于想要比较好的安装体验的同学,我建议不要使用Anaconda3,因为 没那个必要。还很麻烦。

Python3.5.2下载地址:点击打开链接

但是在安装Python3.5的时候,我们还是要注意一些细节工作:

1.勾选 添加路径选项框;

2.用custom的模式安装,选择安装好pip;


这些操作都OK之后,请打开命令行(cmd)或者powershell,不知道的同学请自己在start搜索栏搜;

 >pip install tensorflow-gpu



好啦,这样基本工作就完成啦。由于国内对谷歌限制,可能需要翻墙,连接vpn后下载。

如果不行,请下载相应版本的.whl文件,进行手动操作。不知道的同学请百度。


原创粉丝点击