个体网站建设/软文广告成功案例
近日终于在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 验证是否安装成功。
如果出现如下图所示则表示安装成功。
到此全部安装完成,如在安装过程中遇到问题可以相互交流学习。