Windows10上安裝TensorFlow GPU套件

Windows 10 + Nvidia GeForce GTX 1660 Ti顯示卡,讓tensorflow在GPU上執行

Step1)安裝Anaconda


Step2)安裝Microsoft Visual Studio,筆者安裝Microsoft Visual Studio 2019 Community

在此網址https://visualstudio.microsoft.com/zh-hant/downloads/,下載Microsoft Visual Studio,安裝時勾選安裝C++開發環境。


Step3)安裝NVIDIA CUDA ToolKit,CUDA為Nvidia的平行運算架構,可以有效運用GPU的運算能力,筆者下載11.4.1版本

在此網址https://developer.nvidia.com/cuda-toolkit-archive,選擇CPU架構與作業系統版本,下載程式並安裝


Step4)下載與解壓縮NVIDIA CuDNN((CUDA Deep Neural Network library)),CuDNN為GPU版本的深路學習函式庫,筆者下載CuDNN8.2.2 for CUDA11.4。

在此網址https://developer.nvidia.com/rdp/cudnn-download,登入Nvidia網站下載NVIDIA CuDNN,解壓縮檔案


Step5)設定系統路徑變數

解壓縮NVIDIA CuDNN後發現有3個資料夾,分別是bin、lib與include,複製到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

在系統變數Path下新增以下路徑,

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\libnvvp


Step6)安裝tensorflow-gpu

#pip install tensorflow-gpu


Step7)測試是否安裝完成,執行以下程式是否獲得兩個True。如果是,表示安裝成功。

import tensorflow as tf

tf.test.is_built_with_cuda()

tf.test.is_gpu_available()