ML训练过程不在GPU上

时间:2017-12-23 09:13:26

标签: tensorflow machine-learning google-cloud-platform tensorflow-gpu

由于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

1 个答案:

答案 0 :(得分:1)

也许您安装了tensorflow的cpu版本?

现在,Google Cloud的计算引擎已经安装了VM OS映像,并安装了所有必需的软件,以便更轻松/更快速地入门:https://cloud.google.com/deep-learning-vm/