InternalError: stream did not block host until done; was already in an error state

Viewed 3441

stream did not block host until done; was already in an error state how to fix it using tensorflow keras

model.compile(
optimizer="adam",
loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
metrics=['acc'])    
model.fit(X_train_scaled, y_train, epochs=5)

this error appears

enter image description here

enter image description here

1 Answers

Restarting the notebook fixed this issue for me. If this doesn't fix it, you can try updating the GPU drivers.
Also, a similar issue is available on GitHub : Link to the Issue

Related