search

castacks / daa_docker

Some docker related files for the DAA project.
0 stars 0 forks source link

NGC PyTorch might not work on Jetson #5

Open huyaoyu opened 1 year ago

huyaoyu commented 1 year ago

Tests on Jetson are not successful. It seems like the NGC PyTorch does not work properly on Jetson. See the following post I created in the Nivida forum. https://forums.developer.nvidia.com/t/can-the-ngc-pytorch-images-be-used-on-jetson/237870

I also requested a new feature to the Torch-TensoRT team, asking them to embed Torch-TensoRT in NGC L4T PyTorch images. https://github.com/pytorch/TensorRT/issues/1564

One solution is to compile Torch-TensorRT manually with NGC L4T PyTorch image on a Jetson device.

peri044 commented 1 year ago

We are planning to release a Jetson compatible torch_tensorrt release soon in early January.