All.
This is my simple GPU code in PyTorch.
How can I change this in CPU code?
filename=r’./test/bees/1.jpg’
img = skimage.io.imread(filename)
x = V(centre_crop(img).unsqueeze(0), volatile=True).cuda()
model = models.dict(’resnet18’)
model = torch.nn.DataParallel(model).cuda()
model = torch.load(‘model5.pth’)
logit = model(x)
print(logit)
Thanks in advance .
Enter this code before you call
.cuda()
methodand replacing every
.cuda()
with.to(device)
Great answer from here