安裝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


分享到:


相關文章: