Ошибка выполнения: ошибка cuDNN: CUDNN_STATUS_NOT_SUPPORTED

Question

В colab pytorch cudann показывает проблему

  File "main.py", line 88, in train
    loss.backward()
  File "/usr/local/lib/python2.7/dist-packages/torch/tensor.py", line 195, in backward
    torch.autograd.backward(self, gradient, retain_graph, create_graph)
  File "/usr/local/lib/python2.7/dist-packages/torch/autograd/__init__.py", line 99, in backward
    allow_unreachable=True)  # allow_unreachable flag
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_SUPPORTED. This error may appear if you passed in a non-contiguous input.