我目前正在使用keras和tensorflow训练(微调)一个预先训练好的MobileNet模型。训练是在我的本地计算机上使用GTX980完成的。为了加快速度,我使用基于Ubuntu ()的Amazon Deep Learning AMI在亚马逊网络服务上创建了一个p3.2xlarge实例。当运行一些测试数据(大约300张图片)时,我注意到我<e
我有cuda8.0.61,tensorflow_gpu版本和keras。我在224*224图像数据上训练一个20层的keras模型当我在终端上运行nvidia -smi时,我发现内存正在耗尽,并且计算util的百分比更小。当我尝试拟合模型时,机器变得非常慢。我知道,为了使用gpu并在设备之间切换,我应该使用以下代码: with K.tf.device('/<em
我正在使用Keras和Mac对虚拟数据进行MobileNet体系结构的培训。我同时设置了nump.random和tensorflow.set_random_seed,但由于某些原因,我无法获得可重复的结果:每次我重新运行代码时,都会得到不同的结果。为什么?这不是由于GPU,因为我运行在一个MacBook Pro 2017,其中有一个Radeon显卡,因此Tensorflow没有利用它。编辑:在导入<
我在一台安装并启用了图形处理器的机器上使用了,它似乎只在预测阶段使用了CPU。我在.predict()部分看到的设备放置日志如下:
2020-09-01 06:08:19.085400: I tensorflow/core/common_runtime/eager/execute.ccexecute.cc:573] Executing op ModelDataset in device /job:localhost/replica:0/