Jock Chang
Nov 26, 2018 · 5 min read

由於各軟體版本的更迭,Windows 上安裝GPU版本的Tensorflow也越來越麻煩,這裡記錄下目前安裝的步驟供大家參考。

NVIDIA GPU driver

請檢查裝置管理員中顯示卡的部分是否有NVIDIA GPU?是否有出現問題(驚嘆號)?也可以安裝 NVIDIA GeForce Experience 來自動更新driver

Python

為了讓套件易於管理,建議使用Anaconda安裝Python的開發環境,目前最新的Anaconda v5.3 內附的是 Python v3.7,但 tensorflow 只支援到Python v3.6,可以在這裡下載Anaconda v5.2,以便使用Python 3.6版。下載後也是請以預設值完成安裝(也就是下一步,下一步按到底)。

裝好後,程式集裡面會多出一個 Anaconda Prompt 的項目,切記,之後需要使用 terminal 安裝或執行套件的時候請使用 Anaconda Prompt,不要使用原本 Windows 的 Command prompt,才不會有找不到東西的狀況發生。

使用Anaconda安裝tensorflow-gpu

如果有安裝過前版本的tensorflow,使用下列指令移除

pip uninstall tensorflow-gpu

再以Anaconda安裝

conda install tensorflow-gpu

最後可以用這行指令驗證是否安裝成功:

python -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))"

以下官方安裝流程會出現import error,就可以跳過,目前先留著當備份

參考:https://github.com/tensorflow/tensorflow/issues/22794

Visual Studio 2017/2015

接下來要安裝的 CUDA toolkit 和 tensorflow 都會使用到 Visual studio,因此我們先把他裝好吧

下載 Visual Studio 2017

下載其他舊版的 Visual Studio

請注意,如果是安裝2017版的話,必須自行選擇安裝 components for C++ development 和 Windows 10 SDK (Version 10.0.15063.0)

NVIDIA CUDA Toolkit

原先tensorflow只支援到CUDA 9.0版,目前已支援到CUDA 10.1版

下載 CUDA Toolkit 10.1

如上圖所示,選擇作業系統與版本,然後請按紅框中的Download按鈕開始下載。待下載完畢請執行程式並以預設值完成安裝(也就是下一步,下一步按到底)。

這時大家熟悉的指令 nvidia-smi 應該會出現在 C:\Program Files\NVIDIA Corporation\NVSMI 的資料夾裡,可以打開 command prompt,cd 到 C:\Program Files\NVIDIA Corporation\NVSMI 執行 nvidia-smi 試試看。

NVIDIA cuDNN

下載 NVIDIA cuDNN

登入Nvidia developer後,勾選同意條款,再選擇對應CUDA版本的cuDNN

選擇上圖紅框中版本下載並且解壓縮,會得到一個 cuda的目錄,打開它,把裡面三個目錄複製到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1 便完成安裝。

Tensorflow-GPU

參考: 官方安裝指南

請打開 Anaconda prompt (如果前面是選擇安裝給這台電腦的所有使用者,請按右鍵用系統管理員權限打開 Anaconda prompt)

輸入下列指令:

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.13.1-cp36-cp36m-win_amd64.whl

如果是安裝給單一使用者或有權限問題,請輸入下列指令:

pip install --user --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.13.1-cp36-cp36m-win_amd64.whl

最後可以用這行指令驗證是否安裝成功:

python -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))"

參考:https://towardsdatascience.com/python-environment-setup-for-deep-learning-on-windows-10-c373786e36d1

Taiwan AI Academy

news, tech reviews and supplemental materials

Jock Chang

Written by

Taiwan AI Academy

news, tech reviews and supplemental materials

Welcome to a place where words matter. On Medium, smart voices and original ideas take center stage - with no ads in sight. Watch
Follow all the topics you care about, and we’ll deliver the best stories for you to your homepage and inbox. Explore
Get unlimited access to the best stories on Medium — and support writers while you’re at it. Just $5/month. Upgrade