ubuntu18.04 安装 tensorflow-gpu

  • 当然先要装好 显卡驱动

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

$ nvidia-smi 

测试安装是否成功,接着(装的 python3)

$ pip3 install --upgrade tensorflow-gpu

最后用脚本

import tensorflow as tf

测试下

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