Machine Learning Containers for Jetson

Hi All,

We’ve released the following framework containers for Jetson and JetPack 4.4 Developer Preview on NVIDIA GPU Cloud (NGC):

  • TensorFlow Container (l4t-tensorflow) - contains TensorFlow pre-installed in a Python 3.6 environment to get up & running quickly with TensorFlow on Jetson.

  • PyTorch Container (l4t-pytorch) - contains PyTorch and torchvision pre-installed in a Python 3.6 environment to get up & running quickly with PyTorch on Jetson.

  • Machine Learning Container (l4t-ml) - contains TensorFlow, PyTorch, JupyterLab, and other popular ML and data science frameworks such as scikit-learn, scipy, and Pandas pre-installed in a Python 3.6 environment.

You can also find the Dockerfiles and build scripts on GitHub - have fun!

1 Like