- 當然先要裝好 顯卡驅動
可以用命令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
測試下