Windows 10安裝TensorFlow之CUDA篇

Windows 10安裝TensorFlow之CUDA篇

TensorFlow使用GPU版本效率會更高，本文將講解TensorFlow-GPU版本的安裝，由TensorFlow官網可知安裝GPU版本時所需要的硬件和軟件的requirements，它需要NVIDIA的GPU（硬件）和NVIDIA 驅動、CUDA、CUDNN（軟件），其他可以不用。

接下來將講解CUDA的安裝。

版本對應

前提：你有支持CUDA的NVIDIA GPU

1. 查看TensorFlow版本對應：

   國內TensorFlow官網GPU版本對應關係圖

   國外TensorFlow官網GPU版本對應關係圖

   TensorFlow版本	Python 版本	cuDNN	CUDA
   TensorFlow >= 1.13.0	3.5-3.6	cuDNN >= 7.4.1	CUDA 10.0

   <注：國外的需要梯子，2019/10/12查看和國內一樣的>

2. 安裝最新版本的驅動（會很重要），官網下載即可。

   所有NVIDIA顯卡版本的驅動下載地址：驅動下載

   NVIDIA GeForce 顯卡去驅動下載地址：驅動下載

3. 查看驅動中的系統信息，找支持的CUDA的版本

   a. 桌面（右鍵）–> NVIDIA 控制面板–> 幫助–> 系統信息–> 顯示–> 在細節中查看驅動程序版本

   b. 到NVIDIA官網查看驅動程序對應的CUDA的版本–> CUDA Driver

4. 查看TensorFlow-GPU版本要求
   a. 去TensorFlow官網查看GPU版本要求–> Software requirements

   注：此時可以匹配你要的TensorFlow版本了一般版本1到版本2在NVIDIA大部分顯卡上都可以運行了

由此就知道你的TnsorFlow、CUDA、cuDNN的版本了

安裝

按照前面對應的版本去下載即可

1. GPU驅動程序下好之後，全程下一步即可。

2. CUDA安裝，去官網下載所有版本下載地址，全程下一步即可。

3. cuDNN安裝，所有版本下載地址(需要註冊賬號)，cuDNN安裝時需要複製到CUDA的安裝目錄下（默認位置：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0），將cuDNN下

   cuDNN解壓目錄	複製方向	CUDA安裝目錄
   bin目錄下內容	--------->	bin目錄下
   include目錄下內容	--------->	include目錄下
   lib\x64目錄下內容	--------->	lib\x64目錄下