Win 10安裝TensorFlow GPU並在Jupyter Notebook和Spyder運行

由於前陣子重新安裝 win10，順便藉這機會把 tensorflow GPU的安裝步驟再紀錄得更完整些。

本文分成以下項目：

1. 確認安裝環境

2. 安裝Anaconda

3. 安裝CUDA和cuDNN

4. 安裝tensorflow GPU

5. 測試tensorflow GPU

6. IDE配置tensorflow-gpu環境 (Jupyter Notnbook、Spyder)

7. 其他設定(雙顯卡切換、相關pip、conda指令)

1. 確認安裝環境

作業系統：Windows 10 -64bit

NVIDIA顯卡：Geforce 940M

GPU版的TensorFlow需多加裝CUDA和cuDNN，先確認顯卡是否支援CUDA。

CUDA GPUs連結

TensorFlow 官方最新訊息(2019.1.21)GPU驅動程式版本至少384.x，CUDA v9.0。本文安裝tensorflow-gpu-1.12.0、CUDA：v9.0 (cuDNN：v7.3.1)

phoyo by Tensorflow GPU support

phoyo by Tensorflow GPU buliding from window

2. 安裝Anaconda

Anconda : Anaconda3–2018.12-Windows-x86_64

第一項不要打勾，安裝後自行加入環境變數(path)

# Users\HP是Anaconda3安裝路徑，依你的路徑修改
C:\Users\HP\Anaconda3

[加入方法]點選電腦右鍵>內容>進階系統設定>環境變數>使用者變數>編輯>開啟環境變數編輯畫面>點新增加入C:\Users\HP\Anaconda3>確定

下面開始都用Anaconda Prompt執行指令操作。

3. 安裝CUDA和cuDNN

注意安裝版本與作業系統，本文 CUDA：v9.0 (cuDNN：v7.3.1) for window10 64bit

下載CUDA，選Base Installer，https://developer.nvidia.com/cuda-downloads

下載cuDNN(申請帳號登入)， cuDNN v7.3.1 Library for Windows 10，https://developer.nvidia.com/rdp/cudnn-download

cuDNN解壓縮後，裡面有三個資料夾，將這三個資料夾複製到剛才安裝CUDA的資料夾覆蓋

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

開啟Anaconda Prompt，測試安裝結果

nvcc -V

安裝後新增系統變數

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

[加入方法]系統變數欄位>新增>輸入變數名稱(CUDA_HOME)>輸入變數值>確定

新增環境變數，加入方式如上述

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin

4. 安裝tensorflow GPU

(1) 創建一個名為tensorflow-gpu且python3.5的虛擬環境

conda create -n tensorflow-gpu pip python=3.5

(2) 開啟虛擬環境

activate tensorflow-gpu

(3) 安裝 tensorflow-gpu

pip install — ignore-installed — upgrade tensorflow-gpu

5. 測試tensorflow GPU

老樣子，Hello一下，Hello TensorFlow!

import tensorflow as tf
hello = tf.constant(‘Hello, TensorFlow!’)
sess = tf.Session()
print(sess.run(hello))

或是

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

若你喜歡這篇文章，歡迎點擊 5下 Like ~

支持作者持續創作與內容分享~

6. IDE配置tensorflow-gpu環境 (Jupyter Notnbook、Spyder)

開啟tensorflow-gpu環境

activate tensorflow-gpu

安裝Jupyter Notnbook

conda install jupyter

安裝Spyder

conda install spyder

<同場加映>安裝 Keras

pip install keras

7. 其他設定(雙顯卡切換、相關pip、conda指令)

Intel+NVIDIA顯卡切換，改成預設NVIDIA顯卡優先

桌面點右鍵>開啟NVIDIA控制面板>管理3D設定>廣域設定>選擇NIVIDIA

其他顯卡切換可參考：https://kknews.cc/zh-tw/digital/38objpg.html

其他指令：

#查看所有虛擬環境 
conda info --envs
#移除 tensorflow 虛擬環境
conda remove --name tensorflow --all

相關文章

Python打包成執行檔後檔案太大?Pyinstaller與Numpy的那些事