jetson-containers icon indicating copy to clipboard operation
jetson-containers copied to clipboard

torch2trt build broken

Open hslr4 opened this issue 9 months ago • 2 comments

The torch2trt build fails since an update of torch2trt that removed torch2trt/converters/interpolate.py.

It seems to work again when lines 15 and 16 are removed from the corresponding Dockerfile here: https://github.com/dusty-nv/jetson-containers/blob/ef2ef3e318f2911548af40f6bfccc9ab18c2daf4/packages/pytorch/torch2trt/Dockerfile#L15

Apperantly the problem this sed command fixed, is fixed in torch2trt itself since a previous commit: https://github.com/NVIDIA-AI-IOT/torch2trt/commit/f0520f0b32b2517b8d246537441f08bb5f245b55#diff-f34a4426e314e2b5c6097e416ede5acbe4abde50448ef570a2999783969c5189

hslr4 avatar May 07 '24 12:05 hslr4

Thanks @hslr4, yes I am getting JP6 GA / L4T R35.3 setup now and will update the dockerfile for latest torch2trt 👍

dusty-nv avatar May 07 '24 14:05 dusty-nv

OK, this is fixed now in https://github.com/dusty-nv/jetson-containers/commit/d5bc44ee9abb36db84b48170f9d354366f83de71

dusty-nv avatar May 07 '24 19:05 dusty-nv

thanks 👍

hslr4 avatar May 27 '24 13:05 hslr4