解决Tensorflow使用CPU报错

来源:互联网 发布:js输出时间格式 编辑:程序博客网 时间:2024/05/17 06:27

解决:

Operation was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/cpu:0 ].


tf.app.Flags.DEFINE_boolean('clone_on_cpu',False,'use CPUs to deploy clones.')

改为:

tf.app.Flags.DEFINE_boolean('clone_on_cpu',True,'use CPUs to deploy clones.')

原创粉丝点击