1. 文章
  2. 文章详情

Windows 平台下载安装 Tensorflow框架

一、官方网站

谷歌tensorflow 国内镜像站

https://tensorflow.google.cn/

安装时,应该参考官方安装教程:

https://tensorflow.google.cn/install/

安装 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 的算力:

https://developer.nvidia.com/cuda-gpus

你可以在桌面 鼠标右击,进入 NVIDIA 控制面板,然后查看你的 GPU 型号,确保你的 GPU 性能满足要求。

下载 CUDA 工具包

不同版本的 Tensorflow 对应的 CUDA 版本不一样,必须严格对应上。下载时,应该注意,不是下载最新 CUDA 版本 ,而是下载 CUDA 历史版本

https://developer.nvidia.com/cuda-toolkit-archive

附录A:在 Windows 平台下载及安装 Tensorflow

在上面的历史版本页面中,找到你需要的版本。

安装 CUDA 工具包

下载 CUDA 9.0 版本,下载的包就是 cuda_9.0.176_windows.exe ,点击安装它就可以了。

安装完以后,你可以新建一个额外的环境变量到系统中(这步你可以不做):

环境变量名:CUDA_PATH
环境变量值:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0

附录A:在 Windows 平台下载及安装 Tensorflow

三、下载及添加 cuDNN 库

如果是安装 CPU 版本的 Tensorflow,那么你不需要添加 cuDNN 库。cuDNN 是 CUDA 深度神经网络(Deep Neural Network library)的一个库,以此使 GPU 支持深度神经网络。

下载 cuDNN 库

下载 cuDNN 库之前,需要注册 NVIDIA 账号,并填写一份简单问卷调查表,方可以下载。下面是 cuDNN 页面 :

https://developer.nvidia.com/rdp/cudnn-download

附录A:在 Windows 平台下载及安装 Tensorflow

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

https://developer.nvidia.com/rdp/cudnn-download

附录A:在 Windows 平台下载及安装 Tensorflow

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

附录A:在 Windows 平台下载及安装 Tensorflow

添加 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

附录A:在 Windows 平台下载及安装 Tensorflow

这样 cuDNN 就成功添加到了 CUDA 中。

四、验证安装

在 cmd 命令行终端中,使用下面命令可以 查看显卡信息

nvidia-smi

这个就是英伟达 GPU 的控制面板(NVIDIA System Management Interface),它会显示 GPU 相关信息。不过,你需要将 nvidia-smi 命令所在的目录,添加到系统环境变量中,才可以使用这个命令。这个命令一般在下面这个目录中:

C:Program FilesNVIDIA CorporationNVSMI

附录A:在 Windows 平台下载及安装 Tensorflow

查看 CUDA 版本

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

附录A:在 Windows 平台下载及安装 Tensorflow

五、下载及安装 Anaconda 开发工具

下载 Anaconda 工具包

Anaconda 官网 下载相应的版本的 Anaconda :

https://www.anaconda.com/download/

不同版本的 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 查看版本:

附录A:在 Windows 平台下载及安装 Tensorflow

六、下载及安装 TensorFlow

下载安装

打开系统的 CMD 命令行终端,通过调用以下命令创建名为 tensorflow 的 conda 环境(名字你可以随便起):

conda create -n tensorflow pip python=3.6

附录A:在 Windows 平台下载及安装 Tensorflow

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

附录A:在 Windows 平台下载及安装 Tensorflow

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

附录A:在 Windows 平台下载及安装 Tensorflow

通过下面命令激活 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

参考:https://www.jianshu.com/p/3fa46f3b75a7

发表评论

登录后才能评论

评论列表(0条)