skydam

Results 3 comments of skydam

Yes I did. Turned out the culprit was my PyTorch installation. I think I had it at overnight/preview, but it needed Stable (2.5.1). Once I reinstalled that one it worked...

Would love to help. Running Ubuntu 22 with RTX 3090. Having trouble with current scripts as it refuses to train on the GPU (only CPU).

Yes, but I had to downgrade Tensorflow to 2.2. Version 2.3 didn't do it for me.