Ошибка «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