Window7 64位 安装TensorFlow(GPU)版本

来源:互联网 发布:java注册登录代码 编辑:程序博客网 时间:2024/05/23 12:13

近日终于在win7上安装成功tensorflow GPU版本,并测试成功!这里将其中遇到的种种问题和解决方法做下记录,以供大家相互交流!

电脑配置:


 平台搭建:python3.5.2(64bit)+cuda8.1+cuDNN5.1+tensorflow(tensroflow最新版)

安装过程:

(1)下载安装 Python3.5+

目前TensorFlow在Windows下只支持Python 3.5版本。下载地址:https://www.python.org/downloads/

下载后安装,并配置环境变量。

(2)下载安装 Anaconda3(4.2) 

需要下载64位版本,下载地址:https://www.continuum.io/downloads#windows

下载后安装,在安装的过程中会自动配置环境变量。

安装完Python和Anaconda后用 cmd 打开终端,输入 python  如下图所示安装成功


(3)安装 tensorflow - cpu

在cmd 终端输入 pip install tensorflow   即可安装(此安装的是最新版本的tensorflow)

在安装的 TensorFlow之后会自动安装一些库,如: numpy, sicpy 等

安装完成后即可在 Python中验证,

输入:import tensorflow as tf

(4)安装 tensorflow - gpu

我的电脑gpu型号:GeForce GTX TITAN X

首先安装 CUDA8  和 cudnn5,即显卡驱动。

a:安装 CUDA8

在官网下载 CUDA8,下载地址:https://developer.nvidia.com/cuda-downloads

下载后点击文件自动安装,同时环境变量也自动配置,使用:nvcc -V 查看是否安装成功。

默认安装路径:c:\Program Files\NVIDA GPU Tookit\CUDA\v8.0

b:安装 cudnn5

在官网下载 cudnn5,下载地址:https://developer.nvidia.com/cudnn

cudnn5 是压缩文件,解压放在任何位置,假设位置 c:\cudnn\cuda, 然后将该目录添到path环境变量。

     将 cudnn5 解压后里面的文件拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0下子目录。

C:\cuda\bin\cudnn64_5.dll —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include
C:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64

安装 tensorflow

在cmd 终端输入 pip install tensorflow-gpu   即可安装(此安装的是最新版本的tensorflow)

安装完成后,在cmd中输入Python 之后输入:import tensorflow as tf 验证是否安装成功。

如果出现如下图所示则表示安装成功。



到此全部安装完成,如在安装过程中遇到问题可以相互交流学习。