由於訓練深度學習模型往往需耗費大量運算資源,因此訓練深度學習模型除了使用CPU之外,目前最主流的方法就是使用GPU訓練,
NVIDIA的GPU利用CUDA技術能用於深度學習的建模加速。
確認GPU是否支援CUDA加速運算及運算能力:
GPU可用型號:https://developer.nvidia.com/cuda-gpus
GPU計算能力(Compute Capability)>= 3.5