RuntimeError: ошибка cuDNN: CUDNN_STATUS_NOT_INITIALIZED с использованием pytorch ⇐ Python

[Anonymous]

Я пытаюсь запустить простой пример кода Pytorch. Он отлично работает с использованием процессора. Но при использовании графического процессора я получаю следующее сообщение об ошибке:

Код: Выделить всё

Traceback (most recent call last):
File "", line 1, in 
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/conv.py", line 263, in forward
return self._conv_forward(input, self.weight, self.bias)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/conv.py", line 260, in _conv_forward
self.padding, self.dilation, self.groups)
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

Я пытаюсь запустить следующий код:

Код: Выделить всё

import torch
from torch import nn
m = nn.Conv1d(16, 33, 3, stride=2)
m=m.to('cuda')
input = torch.randn(20, 16, 50)
input=input.to('cuda')
output = m(input)

Я запускаю этот код в докере NVIDIA с CUDA версии 10.2, а у меня графический процессор RTX 2070

Подробнее здесь: https://stackoverflow.com/questions/665 ... ng-pytorch