根據Tensorflow官方文檔,可以使用的Python版本為3.5到3.7,筆者使用的版本是3.7.9,特別向大家提醒,請不要安裝32位元版本的Python,模型訓練過程中會讀取大量資料到記憶體,4GB的記憶體限制捉襟見肘。
請確保安裝時勾選Add Python 3.* to PATH。
安裝完成後請打開命令提示字元輸入py,檢查是否安裝成功,如果沒有出現代表你可能沒有勾選Add Python 3.* to PATH,請直接重裝,請輸入exit()退出Python。
C:\Windows\system32>py
Python 3.7.9 (tags/v3.7.9:13c94747c7, Aug 17 2020, 18:58:18) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> exit()
使用命令提示字元輸入安裝所需套件。
pip install tensorflow-gpu
上述安裝需要一點時間,在等待的同時請註冊NVIDIA Developer的帳號,我們需要從此處下載提供GPU加速的套件,想使用CPU即可的同學,請移步第5天。
筆者所使用的Tensorflow版本為2.3.0,需要10.1版本的CUDA工具,7.6以上版本的CUDNN加速庫,具有CUDA® Compute Capability 3.5以上的顯卡,高於418.x以上版本的Nvidia驅動程式,你可以使用GPU-Z來確認後面兩點。
請根據你當下的Tensorflow版本決定你所需要的CUDA工具版本,你可以透過下列命令取得當前所安裝Tensorflow的版本號,並透過Tensorflow文檔獲取所需資訊。
pip list