PyTorch NG C Контейнер - PullRequest
       38

PyTorch NG C Контейнер

0 голосов
/ 18 марта 2020

Я пытаюсь использовать контейнер pyTorch NG C для обучения модели. Моя docker версия 19.03. Я вытащил контейнер из NG C и запустил его, как https://ngc.nvidia.com/catalog/containers/nvidia: pytorch . Для обучения модели я использую фреймворк Detectron2 . Я попытался запустить часть своего кода и получил следующую ошибку: ImportError: /opt/conda/lib/python3.6/site-packages/detectron2-0.1-py3.6-linux-x86_64.egg/detectron2/_C.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN6caffe26detail37_typeMetaDataInstance_preallocated_32E любая помощь? Поиск в этой проблеме предполагает, что факел и факел видения не совпадают, для меня это не имеет никакого смысла (поскольку контейнер был извлечен из NG C и должен быть предварительно настроен с этим).

Кстати, в рабочей области / README.md упоминается, что на компьютере уже существует conda env (с именем pytorch-35), но у меня есть только базовый env, если я должен был запустить установку скрипт?

Еще раз спасибо!

...