1. 程式人生 > >Windows10系統tensorflow-gpu安裝

Windows10系統tensorflow-gpu安裝

 

安裝前請確保自己的顯示卡支援gpu加速,支援加速的gpu型號可在下面的連結中查詢。

https://www.geforce.com/hardware/technology/cuda/supported-gpus

一、安裝anaconda

  1.   推薦從清華映象站下載,速度快。
  2.   根據自己系統選擇安裝包,現(18.11.21)推薦Anaconda3-5.3.0及更高版本,因為Tensorflow暫不支援Python3.7。
  3.   安裝時勾選Add Anaconda to my PATH environment variable

 

二、安裝cuda及cudnn

  1. 下載安裝cuda,https://developer.nvidia.com/cuda-downloads
  2. 下載cudnn,https://developer.nvidia.com/cudnn ,(下載cudnn需要註冊賬號,請根據cuda版本選擇相對應cudnn)
  3. 解壓cudnn,將解壓出來的三個資料夾下面的檔案移動到對應的cuda資料夾下(cuda預設安裝路徑為 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0)    
  4. 將cuda的lib和include資料夾新增到環境變數

三、下載安裝tensorflow-gpu

  開啟命令列,執行下面命令。

pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple