使用Keras和TensorFlow后端在GPU上出现问题的问题

Question

我在使用Tensorflow后端时遇到问题。我想训练一个keras.applications.VGG16没有顶层的模型，用两个密集层（1024个节点和2个节点）代替它们来解决二进制分类问题。我训练模型在NVidia Geforce GTX 1050与2GB专用VRAM。在

这里的问题是，当我开始培训时，出现了一个错误：

NotFoundError (see above for traceback): No registered 'Snapshot' OpKernel for GPU devices compatible with node training/SGD/gradients/loss/dense_two_loss/Mean_2_grad/floordiv = Snapshot[T=DT_INT32, _device="/job:localhost/replica:0/task:0/device:GPU:0"](training/SGD/gradients/loss/dense_two_loss/Mean_2_grad/Prod/_289)
        .  Registered:  device='CPU'; T in [DT_INT64]
  device='CPU'; T in [DT_INT32]
  device='CPU'; T in [DT_UINT16]
  device='CPU'; T in [DT_INT16]
  device='CPU'; T in [DT_UINT8]
  device='CPU'; T in [DT_INT8]
  device='CPU'; T in [DT_HALF]
  device='CPU'; T in [DT_BFLOAT16]
  device='CPU'; T in [DT_FLOAT]
  device='CPU'; T in [DT_DOUBLE]
  device='CPU'; T in [DT_COMPLEX64]
  device='CPU'; T in [DT_COMPLEX128]
  device='CPU'; T in [DT_BOOL]

[[Node: training/SGD/gradients/loss/dense_two_loss/Mean_2_grad/floordiv = Snapshot[T=DT_INT32, _device="/job:localhost/replica:0/task:0/device:GPU:0"](training/SGD/gradients/loss/dense_two_loss/Mean_2_grad/Prod/_289)]]

有什么问题吗？我的代码如下：

^{pr2}$

Answer 1

问题解决了。我安装了tensorflow的1.7 nightly版本，却忘了删除它。我安装了1.6版本，它工作了：）