Как избежать "RuntimeError: ошибка в LoadLibraryA" для torch.cat?

Question

Я использую решение pytorch для определения каркаса . Я получаю «RuntimeError: ошибка в LoadLibraryA», когда решение выполняет «forward return torch.cat (output, 1)»

Я не могу предоставить минимальный воспроизводимый пример. Поэтому возникает вопрос: можно ли вызвать просто ошибку типа в библиотеке Microsoft при python ошибках программирования, или это, скорее всего, проблема версии (python, pytorch, CUDA, ...) или ошибка в моя установка?

Я использую windows 10, python 3.8.1 и pytorch 1.4.0.

File "main.py", line 144, in <module>
  main()
File "main.py", line 137, in main
  trainer.train(train_loader, val_loader=None)
File "D:\Dev\Python\Projects\wireframe\wireframe\junc\trainer\balance_junction_trainer.py", line 75, in train
  self.step(epoch, train_loader)
File "D:\Dev\Python\Projects\wireframe\wireframe\junc\trainer\balance_junction_trainer.py", line 176, in step
  ) = self.model(input_var, junc_conf, junc_res, bin_conf, bin_res)
File "D:\Dev\Python\Environment\Environments\pytorch\lib\site-packages\torch\nn\modules\module.py", line 532, in __call__
  result = self.forward(*input, **kwargs)
File "D:\Dev\Python\Projects\wireframe\wireframe\junc\model\inception.py", line 41, in forward
  base_feat = self.base_net(im_data)
File "D:\Dev\Python\Environment\Environments\pytorch\lib\site-packages\torch\nn\modules\module.py", line 532, in __call__
  result = self.forward(*input, **kwargs)
File "D:\Dev\Python\Projects\wireframe\wireframe\junc\model\networks\inception_v2.py", line 63, in forward
  x = self.Mixed_3b(x)
File "D:\Dev\Python\Environment\Environments\pytorch\lib\site-packages\torch\nn\modules\module.py", line 532, in __call__
  result = self.forward(*input, **kwargs)
File "D:\Dev\Python\Projects\wireframe\wireframe\junc\model\networks\inception_v2.py", line 97, in forward
  return torch.cat(outputs, 1)
RuntimeError: error in LoadLibraryA

Answer 1

Попробуйте эту рабочую область: запустите следующий код после факела импорта (должно быть исправлено в 1.5):

import ctypes
ctypes.cdll.LoadLibrary('caffe2_nvrtc.dll')

Answer 2

Эту ошибку удалось избежать, понизив ее до python 3.7.6

Примечание: К сожалению, первый шаг общей обработки (время выполнения 3 дня на моем GPU) создает промежуточные результаты с помощью pickel формат 5, новый в Python 3.8. Поэтому мне нужно либо перезапустить первый шаг на 3 дня, либо найти другое решение. Файлы с промежуточными результатами не могут быть использованы с python 3.7.6