tensorflow 安装_安装tensorflow-gpu 2.0

本文介绍tensorflow 2.0 GPU版的安装，测试。软件包括Windows 10，Anaconda，NVIDIA GPU 驱动，CUDA® Toolkit，cuDNN SDK，tensorflow-gpu 2.0.0。

1. 检查电脑正常安装了windows10，NVIDIA GPU 驱动，CUDA兼容的GPU列表看这里https://developer.nvidia.com/cuda-gpus。
2. 安装CUDA Toolkit，tensorflow 2.0 需要CUDA 10.0，下载地址https://developer.nvidia.com/cuda-10.0-download-archive，链接https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_411.31_win10。安装的时候选择custom全部安装。
3. 安装cuDNN SDK，下载地址https://developer.nvidia.com/cudnn，需要注册一个账号。
4. 安装Anaconda3，版本Anaconda3-2019.07-Windows-x86_64，为了加快下载速度，可以从清华大学开源软件镜像站下载，下载地址https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/，链接https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda2-2019.07-Windows-x86_64.exe。
5. 安装tensorflow，在开始菜单点Anaconda3->Anaconda Prompt，打开命令窗口，输入命令pip install tensorflow-gpu==2.0.0 --upgrade -i https://pypi.tuna.tsinghua.edu.cn/simple

• 测试：在命令窗口输入python，进入python 环境，输入import tensorflow as tf，输入print(tf.__version__)