Hello Pytorch community,
I am just getting started out with Pytorch. I am trying to test a trained model. Following the Pytorch tutorial, I have this set up:
1 total = 0
2 correct = 0
3 with torch.no_grad():
4 for data in val_loader:
5 images, labels = data
6 outputs = model(images)
7 _, predicted = torch.max(outputs.data, 1)
8 total += labels.size(0)
9 correct += (predicted == labels).sum().item()
Where val_loader is a DataLoader(), and the model class inherits from the nn.Module.
However, line 6 throws the following error:
$ CUDA_VISIBLE_DEVICES=2,3 python tools/test_net.py
Illegal instruction (core dumped)
I am using Pytorch version 1.1.0, with a CUDA version of 10.2, and cuda driver version 440.33.01. Any help would be much appreciated!
Thank you in advance for your help!