ubuntu18.04 安裝 tensorflow-gpu

  • 當然先要裝好 顯卡驅動

可以用命令ubuntu-drivers devices查看driver列表
先用命令

$ nvidia-smi 

測試安裝是否成功,接着(裝的 python3)

$ pip3 install --upgrade tensorflow-gpu

最後用腳本

import tensorflow as tf

測試下

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章