1 min readAug 12, 2020
Hi Riyaz ! Change this line device = torch.device('cpu') to device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
Written by Ramsri Goutham
Teaches NLP courses at https://www.learnnlp.academy/ ◆ Building AI SaaS Apps: https://questgen.ai/ and https://supermeme.ai/