PyTorch | NVIDIA NGC
catalog.ngc.nvidia.com › nvidia › containersPyTorch is a GPU accelerated tensor computational framework. Functionality can be easily extended with common Python libraries such as NumPy, SciPy, and Cython. Automatic differentiation is done with a tape-based system at both a functional and neural network layer level. This functionality brings a high level of flexibility and speed as a deep ...
NVIDIA L4T ML | NVIDIA NGC
https://catalog.ngc.nvidia.com/orgs/nvidia/containers/l4t-mlMachine Learning Container for Jetson and JetPack. The l4t-ml docker image 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. These containers support the following releases of JetPack for Jetson Nano, TX1/TX2, Xavier NX, and AGX Xavier: