tensorflow 怎樣確定是cpu 還是 gpu
1 回答
尚方寶劍之說
TA貢獻1788條經(jīng)驗 獲得超4個贊
安裝GPU版本的tensorflow就可以了pip install --upgrade 還需要安裝GPU加速包，下載 CUDA 和 cuDNN 的安裝包了，注意版本號分別是 CUDA 8.0 和 cuDNN 5.1。
如果都安裝成功，運行程序時會自動使用GPU計算
- 1 回答
- 0 關(guān)注
- 2552 瀏覽
添加回答
舉報
0/150
提交
取消