torch2trt icon indicating copy to clipboard operation
torch2trt copied to clipboard

Is there a plan to support python 3.10?

Open jakubhejhal opened this issue 2 years ago • 1 comments

Model export is failing on:

  File "/venv/lib/python3.10/site-packages/torch2trt-0.4.0-py3.10.egg/torch2trt/torch2trt.py", line 310, in wrapper
    converter["converter"](ctx)
  File "/venv/lib/python3.10/site-packages/torch2trt-0.4.0-py3.10.egg/torch2trt/converters/interpolate.py", line 81, in convert_interpolate_trt7
    if not isinstance(scales, collections.Sequence):
AttributeError: module 'collections' has no attribute 'Sequence'

Because it has been dropped from the collections module in python 3.10

jakubhejhal avatar Sep 22 '23 11:09 jakubhejhal

I added a pull request, fixing this issue. Please check it out. It was a simple fix. https://github.com/NVIDIA-AI-IOT/torch2trt/pull/884

teodorToshkov avatar Oct 13 '23 06:10 teodorToshkov