解决相关训练报 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