cuda doesn't work

bayes/models.py

@@ -42,7 +42,7 @@ def process_imagenet_get_model(data):
     # Get the vgg16 model, used in the experiments
     model = models.vgg16(pretrained=True)
     model.eval()
-    model.cuda()
+    # model.cuda()
 
     xtest = data['X']
     ytest = data['y'].astype(int)
@@ -72,11 +72,12 @@ def process_imagenet_get_model(data):
                         perturbed_image[segments==i, 1] = background
                         perturbed_image[segments==i, 2] = background
                 perturbed_images.append(transf(perturbed_image)[None, :])
-            perturbed_images = torch.from_numpy(np.concatenate(perturbed_images, axis=0)).float().cuda()
+            perturbed_images = torch.from_numpy(np.concatenate(perturbed_images, axis=0)).float()
             predictions = []
             for q in range(0, perturbed_images.shape[0], batch_size):
                 predictions.append(softmax(model(perturbed_images[q:q+batch_size])).cpu().detach().numpy())
             predictions = np.concatenate(predictions, axis=0)
+            print("Predictions made in models.py:", predictions)
             return predictions
         return wrapped_model
 

data/mnist/mnist_model.py

@@ -94,7 +94,7 @@ def main():
     parser.add_argument('--save-model', action='store_true', default=False,
                         help='For Saving the current Model')
     args = parser.parse_args()
-    use_cuda = True # not args.no_cuda and torch.cuda.is_available()
+    use_cuda = False # not args.no_cuda and torch.cuda.is_available()
 
     torch.manual_seed(args.seed)