I was working on writing a custom training script for my keras GAN model training and when I start training or when I try generating images with my model, I get spammed with this message along with the intended output.
2024-01-23 05:31:31.705379: E ./tensorflow/compiler/xla/stream_executor/stream_executor_internal.h:124] SetPriority unimplemented for this stream.
almost 60 lines of this above message gets displayed when I use my model to train or evaluate or just generate. images I only encounter this problem when using using TPU. When using GPU I dont encounter this message.
Can anyone explain me this issue, why this arises and how to prevent it?
I am having the same issue. Did you find any solution?
The message âSetPriority unimplemented for this streamâ youâre seeing is a common informational warning when using TensorFlow with TPUs, and itâs different from using GPUs. This is due to some features or methods in TensorFlow being implemented for GPUs but not for TPUs. If this message isnât causing any actual failure in your modelâs training or evaluation, itâs likely harmless. To reduce log clutter, you can adjust TensorFlowâs logging level to ERROR:
import tensorflow as tf
tf.get_logger().setLevel(âERRORâ)
This will suppress non-error messages. Keep in mind that TensorFlow is regularly updated, so staying current with the latest version can also help resolve such issues.
1 Like