- 当然先要装好 显卡驱动
可以用命令ubuntu-drivers devices
查看driver列表
先用命令
$ nvidia-smi
测试安装是否成功,接着(装的 python3)
$ pip3 install --upgrade tensorflow-gpu
最后用脚本
import tensorflow as tf
测试下
可以用命令ubuntu-drivers devices
查看driver列表
先用命令
$ nvidia-smi
测试安装是否成功,接着(装的 python3)
$ pip3 install --upgrade tensorflow-gpu
最后用脚本
import tensorflow as tf
测试下