Let’s say your model’s name is net to train on gpu you must have written net. cuda().
After training transfer the model to cpu
net.cpu()
Save the model using torch.save
Load the model using torch.load