Error when I run main.py in MNIST Convnets.
root@albert-PC:/media/ww/examples-master/mnist# python main.py
Traceback (most recent call last):
File "main.py", line 113, in <module>
train(epoch)
File "main.py", line 83, in train
output = model(data)
File "/usr/local/lib/python2.7/dist-packages/torch/nn/modules/module.py", line 206, in __call__
result = self.forward(*input, **kwargs)
File "main.py", line 62, in forward
x = F.relu(F.max_pool2d(self.conv1(x), 2))
File "/usr/local/lib/python2.7/dist-packages/torch/nn/modules/module.py", line 206, in __call__
result = self.forward(*input, **kwargs)
File "/usr/local/lib/python2.7/dist-packages/torch/nn/modules/conv.py", line 237, in forward
self.padding, self.dilation, self.groups)
File "/usr/local/lib/python2.7/dist-packages/torch/nn/functional.py", line 40, in conv2d
return f(input, weight, bias)
RuntimeError: CUDNN_STATUS_ARCH_MISMATCH
GPU: GeForce 710M
CUDA: 8.0