RuntimeError: несоответствие версии cuDNN: PyTorch был скомпилирован для 7102, но связан с 7604 - PullRequest
Новая
       36

RuntimeError: несоответствие версии cuDNN: PyTorch был скомпилирован для 7102, но связан с 7604

0 голосов
/ 10 февраля 2020

Я получил эту ошибку при обучении модели глубокого обучения, и хотя я смотрел на многие решения через Inte rnet, они мне не помогли. Журнал выглядит следующим образом: 

Traceback (most recent call last):
  File "main.py", line 208, in <module>
    main()
  File "main.py", line 100, in main
    model = nn.DataParallel(model).cuda()
  File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/nn/parallel/data_parallel.py", line 105, in __init__
    self.module.cuda(device_ids[0])
  File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/nn/modules/module.py", line 249, in cuda
    return self._apply(lambda t: t.cuda(device))
  File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/nn/modules/module.py", line 176, in _apply
    module._apply(fn)
  File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/nn/modules/rnn.py", line 112, in _apply
    self.flatten_parameters()
  File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/nn/modules/rnn.py", line 78, in flatten_parameters
    if not any_param.is_cuda or not torch.backends.cudnn.is_acceptable(any_param):
  File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/backends/cudnn/__init__.py", line 87, in is_acceptable
    if _libcudnn() is None:
  File "/home/dexter/miniconda3/envs/VideoSum/lib/python3.5/site-packages/torch/backends/cudnn/__init__.py", line 58, in _libcudnn
    'but linked against {}'.format(compile_version, __cudnn_version))
RuntimeError: cuDNN version mismatch: PyTorch was compiled against 7102 but linked against 7604

Ответы [ 2 ]

2 голосов
/ 10 февраля 2020

Вопрос в том, какие версии указаны в журнале, 7.6.4 - это моя версия cudnn, а 7.1.2 - это версия cudnn, исходный код которой скомпилирован. Что мне нужно, это просто понизить (или обновить мою текущую версию cudnn) на: 

conda install cudnn=7.1.2

Это работает, если таковые имеются, пожалуйста, исправьте меня.

1 голос
/ 01 мая 2020

У меня была похожая проблема ... Понижение cudnn работало на меня. Как уже упоминалось, сообщение об ошибке RuntimeError: cuDNN version mismatch: PyTorch was compiled against 7102 but linked against 7604

... указывает, что Pytorch был связан с более новой версией библиотеки cudnn. Это только вызвало проблему в моей модели. Я предполагаю, что это было что-то вроде архаичной c функции Pytorch, которая недавно не обновлялась. После понижения cudnn до той же версии, что и в Pytorch, он снова заработал ... никаких других заметных потерь в производительности.

conda install cudnn=7.1.2

Извините, что повторяюсь, но мой представитель StackOverflow слишком мал добавить комментарий.

...