安裝GPU版本的tensorflow就可以了pip install --upgrade /tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl
還需要安裝GPU加速包,下載 CUDA 和 cuDNN 的安裝包了,註意版本號分別是 CUDA 8.0 和 cuDNN 5.1。
如果都安裝成功,運行程序時會自動使用GPU計算
上一篇:請問人工智能計算器(安卓手機版)怎樣輸入和計算度分秒下一篇:三洋日立貼片機壹直重啟怎麽回事