安装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
安装tensorflow-gpu 2.0
  • 测试:在命令窗口输入python,进入python 环境,输入import tensorflow as tf,输入print(tf.__version__)
安装tensorflow-gpu 2.0


分享到:


相關文章: