Что можно сделать, если возникает ошибка 'There's no data transfer registered for copying tensors' при работе с ONNX?

Ошибка «There's no data transfer registered for copying tensors» при работе с ONNX может возникать из-за неправильной установки ONNX Runtime. nvidia.github.io

Несколько возможных решений проблемы:

• Проверить соответствие версий. stackoverflow.com Нужно убедиться, что версии CUDA, PyTorch и ONNXRuntime, предоставляемые Docker-образом, совпадают с вручную установленными пакетами Python. stackoverflow.com
• Удалить лишнюю установку. stackoverflow.com Возможно, в Dockerfile установлена дополнительная версия ONNXRuntime, которая не поддерживает GPU. stackoverflow.com В таком случае можно попробовать перестроить контейнер без ненужной установки библиотеки. stackoverflow.com
• Использовать ручную установку. nvidia.github.io Например, при использовании CUDA 12 нужно вручную установить ONNX Runtime по инструкциям из документации. nvidia.github.io

Если самостоятельно решить проблему не удаётся, можно обратиться за помощью к специалистам или открыть соответствующую проблему, например, на GitHub. github.com

Пользователь

Время добавления