I tried executing the script model.py from the tensorrt sample python directory
/usr/src/tensorrt/samples/python/network_api_pytorch_mnist
$ python3.6 model.py
It is throwing this message
/home/mistral/.local/lib/python3.6/site-packages/torchvision/io/image.py:11: UserWarning: Failed to load image Python extension:
** warn(f"Failed to load image Python extension: {e}")**
Let me know what could be the reason.
Thank you