Error in onnx_export.py

Hi Dusty,
I am getting the following error while I am exporting using the onnx_export.py
This is the cmd I used :
python3 onnx_export.py --model-dir=models/model01/ --labels=models/model01/labels.txt

Traceback (most recent call last):
File “onnx_export.py”, line 9, in
from vision.ssd.vgg_ssd import create_vgg_ssd
File “/home/anupriya/jetson-inference/python/training/detection/ssd/vision/ssd/vgg_ssd.py”, line 6, in
from .predictor import Predictor
File “/home/anupriya/jetson-inference/python/training/detection/ssd/vision/ssd/predictor.py”, line 4, in
from .data_preprocessing import PredictionTransform
File “/home/anupriya/jetson-inference/python/training/detection/ssd/vision/ssd/data_preprocessing.py”, line 1, in
from …transforms.transforms import *
File “/home/maanupriyayank/jetson-inference/python/training/detection/ssd/vision/transforms/transforms.py”, line 5, in
from torchvision import transforms
File “/home/anupriya/.local/lib/python3.8/site-packages/torchvision/init.py”, line 6, in
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils
File “/home/anupriya/.local/lib/python3.8/site-packages/torchvision/_meta_registrations.py”, line 4, in
import torch._custom_ops
File “/home/anupriya/anaconda3/lib/python3.8/site-packages/torch/_custom_ops.py”, line 3, in
from torch._custom_op.impl import (
File “/home/anupriya/anaconda3/lib/python3.8/site-packages/torch/_custom_op/impl.py”, line 9, in
from torchgen.model import FunctionSchema, OperatorName, SchemaKind, BaseType, ListType, BaseTy
File “/home/anupriya/anaconda3/lib/python3.8/site-packages/torchgen/model.py”, line 2354, in
import torchgen.api.ufunc as ufunc
File “/home/anupriya/anaconda3/lib/python3.8/site-packages/torchgen/api/ufunc.py”, line 11, in
import torchgen.api.types as api_types
File “/home/anupriya/anaconda3/lib/python3.8/site-packages/torchgen/api/types/init.py”, line 1, in
from .types import *
File “/home/mayank/anaconda3/lib/python3.8/site-packages/torchgen/api/types/types.py”, line 99, in
ScalarType.Float8_e5m2: float8_e5m2T,
File “/home/mayank/anaconda3/lib/python3.8/enum.py”, line 384, in getattr
raise AttributeError(name) from None
AttributeError: Float8_e5m2

Hi,

Could you share the below info with us first?

>>> import torch
>>> torch.cuda.is_available()
>>> torch.__version__
>>> import torchvision
>>> torchvision.__version__

Thanks.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.