Tensorflow学习——Win10安装Tensorflow-gpu和CUDA及cuDNN

Tensorflow安装官方链接

cuda安装官方链接

一、前期准备

1. 检查电脑GPU是否达到要求，否则安装CPU版本，检查网站：https://developer.nvidia.com/cuda-gpus
2. 安装Visual C++，若安装了Visual Studio则可不管这项

方式一（安装python，并在python环境下安装tensorflow） 

• 安装python3，下载网站：https://www.python.org/downloads/ 
• 安装numpy

方式二（安装Anaconda，并在Anaconda环境下安装tensorflow，建议采用这种方式）

• 安装Anaconda3，下载网站：https://www.anaconda.com/download/#windows
• 创建tensorflow环境

 

二、安装CUDA和cuDNN

因Tensorflow-gpu对CUDA和cuDNN版本有所要求，根据安装的Tensorflow-gpu版本下载对应CUDA和cuDNN版本，可参照下图。

cuda下载链接：https://developer.nvidia.com/cuda-downloads

cudnn下载链接：https://developer.nvidia.com/rdp/form/cudnn-download-survey（需要注册填调查问卷）

下载后默认安装即可，安装完成后将目录\CUDA\v9.0\bin添加到系统路径Path中。

 

三、安装Tensorflow-gpu

方式一：

    本文采用pip的方式安装，打开Windows Powershell（建议打开管理员模式），运行下面命令，默认安装最新版本Tensorflow-gpu，若速度慢可以改用镜像。

pip3 install --upgrade tensorflow-gpu

方式二：

（1）打开Anaconda Prompt，输入以下命令创建独立环境，该环境与Anaconda独立。

conda create -n tensorflow-gpu python=3.6

 备注1：若出现错误 SafetyError，运行以下命令清空环境

conda env remove --tensorflow-gpu        # Remove the environment
conda clean --packages --tarballs    # Clean unused cached packages.

  备注2：若下载缓慢可改用镜像

conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

（2）加载环境

conda activate tensorflow-gpu

（3）安装

pip install --upgrade tensorflow-gpu

 

四、测试

    打开Windows Powershell，输入下面命令。若采用Anaconda，则运行Spyder(tensorflow-gpu)。

python

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))

若获得以下结果则为安装成功。

Hello, TensorFlow!

常见问题解决方法：https://www.tensorflow.org/install#common_installation_problems