Setup corectly on GPU

Hi. :slight_smile:
How can i see if i made the correct setup on the GPU?

I am also learning about that. I found that
Did you find any good resources? I thought there can be more guidance in part 2. I am still searching it. I tried Colab, which I don’t need to do much in set it up, just change the runtime type, but with issues in something like widget. I got stuck in lesson 2 because of that.

Now I’m using this to see if i use GPU or CPU:

device = torch.device(‘cuda’ if torch.cuda.is_available() else ‘cpu’)

I don’t know if what I’m doing is right, but for me it is working.