That’s what I have done. I’ve been using Keras3 with PyTorch within a WSL environment since beginning Feb (when I was unable to get the build from source working). I’ve also noticed that they have now released a nightly PyTorch build with RTX50xx support for Windows, but my WSL environment is all set up already so not going to switch to that soon.
Thank you for pointing out that the container only has TF2.17.1 installed. I overlooked that…
So it looks like I’ll stick to the nightly PyTorch for now then.
One thing that isn’t working in Keras with PyTorch is stateful LSTM networks. I picked that up “Modified by moderator” and worked on a solution that has now been approved. If you need that, just follow the colab link and copy the new LSTM class he created for use in your project before the change is released Fix PyTorch stateful RNN/LSTM gradient computation error resolves #20875 by praveenhosdrug123 · Pull Request #20916 · keras-team/keras · GitHub