重要提示
- TF 2.10 是 Windows 平台上 原生環境最後一版可以支援 GPU 的版本.
- 如果需要在 Windows 中使用 TF2.11 及以後的版本, 請改用WSL 2 環境.
前言
這個主題大概是所有平時只用 Windows 的人士心中永遠的痛 (尤其是不懂系統, 或者不十分了解系統的人). 其實 Linux 用戶也好不到哪裡去啦 (如果你不是 Linux 上系統工程師), 差別只是無解和有點希望. 不過這個問題在 Win10/Win11 上現在幾乎已經完全解開, 不再是無解的問題了.
目前在 Win10/Win11 平台上, 要啟用 GPU 對於 TensorFlow 的支援, 可用的作法可以分為四類:
- 使用 Windows 原生環境 (及 Python 虛擬環境)
- 使用 WSL2 環境
- 使用 docker 環境
- 使用 Anaconda for Windows 環境
不過在開始安裝之前, 各位有二件事要決定一下: