3.1 環境配置
在開始本章后面的學習之前,我們先簡單介紹本章必要的軟件安裝與配置,主要包括下面的軟件包安裝。
? 安裝Python環境:本書的代碼已在Python 3.7中編譯通過,讀者可以在Anaconda官網中(鏈接3-1),根據自己的操作系統平臺選擇對應的安裝版本,參見圖3-1。

圖3-1 Anaconda官網安裝界面
? GPU環境配置(可選):如果訓練中使用的模型是深度學習模型,建議讀者使用帶有GPU的設備來提升模型訓練的速度。為了使深度學習框架支持GPU編程,需要首先安裝CUDA和cuDNN。
CUDA安裝:在官網(鏈接3-2)下載與操作系統匹配的版本并安裝,參見圖3-2。

圖3-2 CUDA安裝界面
cuDNN安裝:在官網(鏈接3-3)找到與CUDA相兼容的版本進行下載并安裝,參見圖3-3。
? 安裝PyTorch:在安裝Anaconda后,我們可以直接使用pip來安裝PyTorch。使用pip的好處是,系統能夠自動檢測出合適的PyTorch版本,并自動安裝依賴庫。直接在命令行中輸入下面的命令即可。

pip
install
torch

圖3-3 cuDNN安裝界面