Cómo instalar onnx correctamente para nvidia jetson nano

Introducción

Este artículo no necesita compilar pybind11 desde el código fuente . Busqué en el sitio antes y compilé pybind11 desde el código fuente. Es un rumor. Aquí hay dos artículos llamados Rumor 1 y Rumor 2 .
En primer lugar, la versión 1.12.0 de onnx es incompatible con la versión 3.0.0 del compilador protobuf instalada en ubuntu18.04 en jetson nano . Debido a que se debe ejecutar Yolov5, probé el requisito mínimo, que es el 1.9.0 versión de onnx , y la instalé con éxito. Además, todo el pip en este artículo se refiere al pip de python3.6.9. No tengo el pip de python2 aquí. Si aún lo tiene, reemplace pip con pip3.

Pasos de instalación

No usaré sudo para obtener permisos. Al instalar paquetes, use un entorno virtual o -u corte los permisos de los usuarios para instalar, o sudo primero como yo, y luego instale todo directamente bajo permisos de raíz (no recomendado).

1. Instalar protobuf relacionado

apt-get install protobuf-compiler libprotobuf-dev

Cuando se instala onnx, buscará el compilador protobuf existente, por lo que debe instalarse primero; de lo contrario, se informará el error "no se encontró el compilador protobuf onnx".

2. Instalar pybind11

Ejecutar pip install pybind11 directamente, no funciona, pero:

pip install pybind11[global]

Así es, siempre que se agregue [global] después de pybind11, se resolverá el error
"onnx no pudo encontrar pybind11 faltando pybind11_DIR" o la falta de "pybind11Config.cmake", "pybind11-config.cmake" .

3. Instalar onnx

pip install onnx==1.9.0

¿Por qué es esta versión? La mencioné en la introducción; de lo contrario, se informará el error "onnx error no match for operator []".

exhibición final

Ponga una foto de una instalación exitosa, para que nadie diga palabras vacías.

inserte la descripción de la imagen aquí

Supongo que te gusta

Origin blog.csdn.net/weixin_43945848/article/details/127224535
Recomendado
Clasificación