一、官方网站
谷歌tensorflow 国内镜像站:
安装时,应该参考官方安装教程:
安装 CPU 版本的 TensorFlow 是比较容易的,这里尝试安装 GPU 版本的 TensorFlow ,具体如下:
- 操作系统:
window 7 64位 - Tensorflow 版本 (GPU版):
r1.10 - GPU 型号:
NVIDIA Quadro GTX750Ti - cuDDN 版本:
Download cuDNN v7.0.5 (Dec 5, 2017), for CUDA 9.0 - CUDA 版本:
9.0 - VScode 版本:
1.27.1 - Anconda 版本:
Anaconda3-5.2.0-Windows-x86_64.exe(一个 Python 3.6的版本) - Python 版本:
Python 3.6
二、下载及安装 CUDA 工具包
如果你安装的是 CPU 版本的 TensorFlow,那么就不需要安装 CUDA。安装 GPU 版本,需要你的电脑有 NVIDIA GPU 才可以,并且要求算力在 3.5 以上,可以在 NVIDIA 官网查看 GPU 的算力:
你可以在桌面 鼠标右击,进入 NVIDIA 控制面板,然后查看你的 GPU 型号,确保你的 GPU 性能满足要求。
下载 CUDA 工具包
不同版本的 Tensorflow 对应的 CUDA 版本不一样,必须严格对应上。下载时,应该注意,不是下载最新 CUDA 版本 ,而是下载 CUDA 历史版本 :

在上面的历史版本页面中,找到你需要的版本。
安装 CUDA 工具包
下载 CUDA 9.0 版本,下载的包就是 cuda_9.0.176_windows.exe ,点击安装它就可以了。
安装完以后,你可以新建一个额外的环境变量到系统中(这步你可以不做):
环境变量名:CUDA_PATH
环境变量值:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0

三、下载及添加 cuDNN 库
如果是安装 CPU 版本的 Tensorflow,那么你不需要添加 cuDNN 库。cuDNN 是 CUDA 深度神经网络(Deep Neural Network library)的一个库,以此使 GPU 支持深度神经网络。
下载 cuDNN 库
下载 cuDNN 库之前,需要注册 NVIDIA 账号,并填写一份简单问卷调查表,方可以下载。下面是 cuDNN 页面 :

注册、登录并填完问卷调查以后,才能进入 cuDNN 下载页面:

我这里下载的是 Download cuDNN v7.0.5 (Dec 5, 2017), for CUDA 9.0 也就是 cuDNN 7.0 版本,对应于 CUDA 9.0 。你可以把安装文档下载下来看看:

添加 cuDNN 库到 CUDA 中
将刚才下载下来的 cuDNN 库进行解压,依次将库文件放入对应文件夹中即可:
cuDNN 相关文件 复制到 CUDA 相应目录
cudnn-9.0-windows7-x64-v7cudabincudnn64_7.dll 复制到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0bin
cudnn-9.0-windows7-x64-v7cudaincludecudnn.h 复制到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0include
cudnn-9.0-windows7-x64-v7cudalibx64cudnn.lib 复制到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0libx64

这样 cuDNN 就成功添加到了 CUDA 中。
四、验证安装
在 cmd 命令行终端中,使用下面命令可以 查看显卡信息:
nvidia-smi
这个就是英伟达 GPU 的控制面板(NVIDIA System Management Interface),它会显示 GPU 相关信息。不过,你需要将 nvidia-smi 命令所在的目录,添加到系统环境变量中,才可以使用这个命令。这个命令一般在下面这个目录中:
C:Program FilesNVIDIA CorporationNVSMI

查看 CUDA 版本
可以通过 nvcc -V 命令,查看 CUDA 版本信息:

五、下载及安装 Anaconda 开发工具
下载 Anaconda 工具包
到 Anaconda 官网 下载相应的版本的 Anaconda :
不同版本的 Anaconda 支持的 Python 版本不一样。对于支持 Python2 的版本,统一以 Anaconda 2 开头。对于支持 Python 3 的版本,统一以 Anaconda 3 开头。
安装 Anaconda 工具包
如果你在 windows 上安装过 python ,为了不让其起冲突,应该先卸载之前安装的 python,将你之前加入系统环境变量 PATH 中的 python 路径去掉:
C:Python27;C:Python27Scripts;C:Program FilesPython36Scripts;C:Program FilesPython36;
在 Windows 系统上,傻瓜式安装即可。我这里安装的是 Anaconda3-5.2.0-Windows-x86_64.exe,是一个 python3.6 的版本。安装完以后,将 anconda 相关路径加入系统环境变量中(具体路径根据你的安装路径而定):
D:ProgramDataAnaconda3
D:ProgramDataAnaconda3Scripts
将上面两个路径加入系统环境变量 PATH 以后,输入 python -V 查看版本:

六、下载及安装 TensorFlow
下载安装
打开系统的 CMD 命令行终端,通过调用以下命令创建名为 tensorflow 的 conda 环境(名字你可以随便起):
conda create -n tensorflow pip python=3.6

这样,就创建了一个名为 tensorflow 的环境,直观的就是在 andconda3 中的 envs 文件夹下,生成一个 tensorflow 文件夹:

进入 tensorflow 这个文件夹下,发现一个 python.exe ,这个就是我们刚才指定的 python=3.6,并且有一些 Lib 库,一会我们下载的 GPU 版本 tensorflow 就会放在 Lib 库中。envs 每个环境都有一个自己的文件夹,各自用各自文件夹内的版本程序,从而到达版本控制和环境隔离。

通过下面命令激活 conda 环境:
activate tensorflow
在 conda 环境中安装 TensorFlow ,这里是 GPU 版本:
pip install --ignore-installed --upgrade tensorflow-gpu
如果你安装的是 CPU 版本,请执行下面命令:
pip install --ignore-installed --upgrade tensorflow
测试是否安装成功
如果你刚才没有退出 tensorflow 的 anconda 环境,那么 tensorflow 环境是处于激活状态的。如果你退出了,可以执行下面命令激活:
activate tensorflow
激活以后,命令行终端前面会有一个 tensorflow 的显示,表明你处于 tensorflow 环境中:
(tensorflow) D:1_wwy-dir4_tensorflow-workspace>
接下来,在这个环境中,使用 shell 中的 python :
python
然后依次输入以下代码命令:
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
如果打印出下面内容,说明 tensorflow 正常:
Hello, TensorFlow!七、错误或者警告处理
下面的命令执行,是在 windows 系统的 cmd 命令行终端中执行的,你也可以使用 vscode 的命令行终端(快捷键 Ctrl + ` 打开 vscode 集成终端)。注意:执行命令之前要激活 tensorflow 环境:
activate tensorflow
因为我们是针对 tensorflow 环境进行安装配置。
setuptools 版本
使用过程,可能会发出下面警告信息:
tensorflow-gpu 1.10.0 has requirement setuptools<=39.1.0, but you'll have setuptools 40.2.0 which is incompatible.
这是叫你进行 setuptools 版本回退,当前安装的 setuptools 工具版本太新,回退到 39.1.0 以前的版本。先卸载 setuptools 工具 :
pip uninstall setuptools
再执行安装命令:
pip install --ignore-installed setuptools==39.1.0
pip 版本升级提醒
如果 pip 版本太老,可能会提示你升级:
You are using pip version 10.0.1, however version 18.0 is available.
You should consider upgrading via the 'python -m pip install --upgrade pip' command.
执行下面命令,升级即可:
python -m pip install --upgrade pip
评论列表(0条)