How to make the code run using the local gpu

[SKBL5694]

I run the notebook on my PC, when I use learn.fintune(), I found that the time consumed is far higher than mentioned in the notes, I found that this is due to training using cpu instead of gpu, because when I run the code, the memory use of GPU doesn't increase. How can I solve the problem?

[CaseyHaralson]

I run the notebook on my PC, when I use learn.fintune(), I found that the time consumed is far higher than mentioned in the notes, I found that this is due to training using cpu instead of gpu, because when I run the code, the memory use of GPU doesn't increase. How can I solve the problem?

Try running this in a notebook to see if CUDA is available. If you get a False output then you need to fix the torch install. Try installing the correct thing from here: https://pytorch.org/get-started/locally/

import torch
torch.cuda.is_available()