I was running TensorFlow 2.15 + Keras 2.15 (training without issue) but was running into an error when loading the model after training. I found out the bug was resolved in TensorFlow 2.17 and Keras 3.4.1. So I decided to upgrade, however once I did, I noticed my GPU (RTX4090) was not being used when training, or at least that’s how it appeared because when monitoring my GPU in the Activity Monitor, it seemed like it wasn’t using my GPU at all, it would run at like 0-3%, while the memory was at 100% and would take 100x the amount of time per epoch. Prior to this, when using TF 2.15 it would take at most 2 minutes per epoch. So I decided to do a clean install and install CUDA Toolkit 12.3 + cuDNN 8.9.7 (as suggested by the TensorFlow Documentation). But found no change.