Missing Deep Learning Framework Support for Thor

Hi,

I’m unable to find compatible wheel files for the following frameworks on NVIDIA Thor:

  • PyTorch

  • TorchVision

  • TensorFlow

I’ve also looked for Docker images that support these frameworks with our configuration (JetPack 7 + CUDA 13.0 + Ubuntu 24.04), but haven’t found suitable options.

System:

  • Device: NVIDIA Thor

  • OS: Ubuntu 24.04.3

  • JetPack: 7

  • CUDA: 13.0

Could someone point me to the correct repositories or resources where I can obtain these frameworks for Thor?

Thanks!

Hi,

Thor uses the SBSA driver instead of the iGPU driver, so the container name is slightly different.

Please find Python packages in the link below:

PyTorch container: nvcr.io/nvidia/pytorch:25.09-py3

Triton Inference server with vLLM backend: nvcr.io/nvidia/tritonserver:25.09-vllm-python-py3

Thanks.

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