當前位置:編程學習大全網 - 電腦編程 - tensorflow 在windows 下怎麽使用gpu

tensorflow 在windows 下怎麽使用gpu

安裝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計算

  • 上一篇:請問人工智能計算器(安卓手機版)怎樣輸入和計算度分秒
  • 下一篇:三洋日立貼片機壹直重啟怎麽回事
  • copyright 2024編程學習大全網