RuntimeError: CUDNN_STATUS_ARCH_MISMATCH

vision
1

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

2

cudnn does not support CUDA arch 2.1 cards. The minimum supported CUDA arch is 3.0

You can find out what CUDA arch your GPU belongs to in this page:

2 Likes