How to activate GPU training?

[usama-muneer]

I tried to find out the GPU training setting. Can you share the code snippet where can I set the GPU for training and testing?

[moh3enrafiei]

you should be better use google colab and active GPU runtime easily

[AniketP04]

This is the code snippet:

def get_device():

    return torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')

I hope this will solve resolve your issues.