解决相关训练报 OpKernel was found, but attributes didn‘t match) Requested Attributes: T=DT_BFLOAT16

解决相关训练报 OpKernel was found, but attributes didn't match) Requested Attributes: T=DT_BFLOAT16

I1102 08:59:56.310849 139771096057664 basic_session_run_hooks.py:606] Saving checkpoints for 0 into /notebooks/efficientnet/model300/model.ckpt.
2020-11-02 09:00:21.343364: E tensorflow/core/common_runtime/executor.cc:641] Executor failed to create kernel. Not found: No registered 'Sigmoid' OpKernel for GPU devices compatible with node {{node Sigmoid}}
	 (OpKernel was found, but attributes didn't match) Requested Attributes: T=DT_BFLOAT16
	.  Registered:  device='XLA_CPU'; T in [DT_FLOAT, DT_DOUBLE, DT_COMPLEX64, DT_BFLOAT16, DT_COMPLEX128, DT_HALF]
  device='XLA_CPU_JIT'; T in [DT_FLOAT, DT_DOUBLE, DT_COMPLEX64, DT_BFLOAT16, DT_COMPLEX128, DT_HALF]
  device='XLA_GPU_JIT'; T in [DT_FLOAT, DT_DOUBLE, DT_COMPLEX64, DT_BFLOAT16, DT_HALF]
  device='XLA_GPU'; T in [DT_FLOAT, DT_DOUBLE, DT_COMPLEX64, DT_BFLOAT16, DT_HALF]
  device='CPU'; T in [DT_COMPLEX128]
  device='CPU'; T in [DT_COMPLEX64]
  device='CPU'; T in [DT_DOUBLE]
  device='CPU'; T in [DT_HALF]
  device='CPU'; T in [DT_FLOAT]
  device='GPU'; T in [DT_DOUBLE]
  device='GPU'; T in [DT_HALF]
  device='GPU'; T in [DT_FLOAT]

	 [[Sigmoid]]
ERROR:tensorflow:Error recorded from training_loop: 2 root error(s) found.
  (0) Not found: No registered 'Sigmoid' OpKernel for GPU devices compatible with node {{node Sigmoid}}
	 (OpKernel was found, but attributes didn't match) Requested Attributes: T=DT_BFLOAT16
	.  Registered:  device='XLA_CPU'; T in [DT_FLOAT, DT_DOUBLE, DT_COMPLEX64, DT_BFLOAT16, DT_COMPLEX128, DT_HALF]

查看相关训练代码   含有bfloat16   设置为 falese