Эта версия TensorRT поддерживает отступы только по двум внешним измерениям.

Я пытаюсь преобразовать файл модели onnx AI в файл nvidia tensorrt. Аппаратное обеспечение: NVIDIA xavier AGX. Версия tenorrt и информация об оборудовании: NVIDIA Jetson AGX Xavier [16 ГБ]

  • Реактивный ранец 4.5.1 [L4T 32.5.1]
  • Энергетический режим NV: MAXN - Тип: 0
  • jetson_stats.service: активен
  • Библиотеки:
    • CUDA: 10.2.89
    • cuDNN: 8.0.0.180
    • TensorRT: 7.1.3.0
    • Visionworks: 1.6.0.501
    • OpenCV: 3.4.15-dev скомпилированный CUDA: НЕТ
    • VPI: ii libnvvpi1 1.0.15 arm64 Библиотека интерфейса программирования NVIDIA Vision
    • Вулкан: 1.2.70

Однако, когда я запускаю код : с trt.Builder (TRT_LOGGER) как b, b.create_network(explicit_batch) как n, trt.OnnxParser(n, TRT_LOGGER), я получил следующую ошибку:

Сборка движка fp32 TensorRT из файла onnx ... Начало синтаксического анализа файла ONNX [TensorRT] ПРЕДУПРЕЖДЕНИЕ: onnx2trt_utils.cpp:220: ваша модель ONNX была сгенерирована с весами INT64, в то время как TensorRT изначально не поддерживает INT64. Попытка сбросить до INT32. При разборе файла onnx обнаружена 1 ошибка.

Файл модели Onnx : Conv pads: 1, 1, 2, 2.

Кто-нибудь знает как исправить?

0 ответов

Другие вопросы по тегам