Hello,
I have defined a U-net as well as training code, however I don’t think it is training with CUDA (using GPU’s). In my limited experience with tf, I know that as I began training with CUDA, it would output a message stating that CUDA is running, devices were identified etc. Is there something similar for pytorch? I am just going based on the fact that GPU utilization is very low.

I have included a few lines in my code that I thought would encourage CUDA use, but I don’t beieve it’s worked ie.
self.device = torch.device("cuda:0")
torch.set_default_tensor_type("torch.cuda.FloatTensor")

Is there anything else for me to keep in mind? I briefly reviewed https://pytorch.org/docs/stable/notes/cuda.html#:~:text=cuda%20is%20used%20to%20set,cuda 1. but I am a bit ‘lost in the documentation’
Thanks,
kyle