由于GPU价格较低,我刚刚从AWS迁移到Google Cloud Platform。我按照网站上的说明创建了一个带有K80 GPU的计算引擎实例,安装了最新版本的Tensorflow,keras,cuda驱动程序和cnDNN,一切都很顺利。但是当我尝试训练我的模型时,训练过程仍在CPU上。
NVIDIA-SMI 387.26 Driver Version: 387.26
Cuda compilation tools, release 9.1, V9.1.85
Tensorflow version -1.4.1
cudnn cudnn-9.1-linux-x64-v7
Ubuntu 16.04
答案 0 :(得分:1)
也许您安装了tensorflow的cpu版本?
现在,Google Cloud的计算引擎已经安装了VM OS映像,并安装了所有必需的软件,以便更轻松/更快速地入门:https://cloud.google.com/deep-learning-vm/