RuntimeError: CUDNN_STATUS_BAD_PARAM

[alphamupsiomega]

What could be the cause and fix of this CUDNN_STATUS_BAD_PARAM error? I cannot get this to extract the features. Pytorch otherwise runs fine on my Ubuntu + Anaconda machine.

my@my:~/Dropbox/x/CV/pytorch-pruning$ CUDA_VISIBLE_DEVICES=0 python finetune_original.py --train Downloading: "https://download.pytorch.org/models/vgg16-397923af.pth" to /home/my/.torch/models/vgg16-397923af.pth 100%|553433881/553433881 [08:01<00:00, 1148236.75it/s] Epoch: 0 Traceback (most recent call last): File "finetune.py", line 268, in if args.train: File "finetune.py", line 163, in train print "Epoch: ", i File "finetune.py", line 181, in train_epoch for batch, label in self.train_data_loader: File "finetune.py", line 176, in train_batch else: File "/home/my/anaconda2/lib/python2.7/site-packages/torch/nn/modules/module.py", line 206, in call result = self.forward(*input, **kwargs) File "finetune.py", line 40, in forward def forward(self, x): File "/home/my/anaconda2/lib/python2.7/site-packages/torch/nn/modules/module.py", line 206, in call result = self.forward(*input, **kwargs) File "/home/my/anaconda2/lib/python2.7/site-packages/torch/nn/modules/container.py", line 64, in forward input = module(input) File "/home/my/anaconda2/lib/python2.7/site-packages/torch/nn/modules/module.py", line 206, in call result = self.forward(*input, **kwargs) File "/home/my/anaconda2/lib/python2.7/site-packages/torch/nn/modules/conv.py", line 237, in forward self.padding, self.dilation, self.groups) File "/home/my/anaconda2/lib/python2.7/site-packages/torch/nn/functional.py", line 40, in conv2d return f(input, weight, bias) RuntimeError: CUDNN_STATUS_BAD_PARAM

[jacobgil]

Still relevant? Try printing x.size() in forward. It should be [batch_size, 3, height, width]