Hi, can you check if you have a CUDA-supported installation of PyTorch? torch.cuda.is_available is sufficient
Thanks @martinkim0 for your reply. Running torch.cuda.is_available returned False. Then I followed the instruction posted here to resolve the issue. Now it is running fine with making use of GPU.
1 Like
Awesome, glad you were able to resolve it!