CatalogContainers
Sort: Last Modified
Triton Inference Server (Formerly TensorRT inference Server)
ContainerTriton Inference Server is an open source software that lets teams deploy trained AI models from any framework, from local or cloud storage and on any GPU- or CPU-based infrastructure in the cloud, data center, or embedded devices.
TensorFlow
ContainerTensorFlow is an open source platform for machine learning. It provides comprehensive tools and libraries in a flexible architecture allowing easy deployment across a variety of platforms and devices.
PyTorch
ContainerPyTorch is a GPU accelerated tensor computational framework. Functionality can be extended with common Python libraries such as NumPy and SciPy. Automatic differentiation is done with a tape-based system at the functional and neural network layer levels.
TensorRT
ContainerNVIDIA TensorRT is a C++ library that facilitates high-performance inference on NVIDIA graphics processing units (GPUs). TensorRT takes a trained network and produces a highly optimized runtime engine that performs inference for that network.
Kaldi
ContainerKaldi is an open-source software framework for speech processing.
TAO Toolkit for Conv AI
ContainerDocker container with workflows implemented in PyTorch as part of the Train Adapt Optimize (TAO) Toolkit.
Merlin Tensorflow Training
ContainerThis container allows users to do preprocessing and feature engineering with NVTabular, and then train a deep-learning based recommender system model with TensorFlow.
Merlin Training
ContainerThis container allows users to do preprocessing and feature engineering with NVTabular, and then train a deep-learning based recommender system model with HugeCTR.
DCGM Exporter
ContainerMonitor GPUs in Kubernetes using NVIDIA DCGM. This is an exporter for a Prometheus monitoring solution in Kubernetes.
CUDA Samples
ContainerCollection of containerized CUDA Samples
NVIDIA Kubernetes Device Plugin
ContainerThe NVIDIA Kubernetes Device Plugin registers GPUs as compute resources in the Kubernetes cluster.
NVIDIA GPU Feature Discovery for Kubernetes
ContainerPlugin for the Kubernetes Node Feature Discovery for adding GPU node labels.
NVIDIA MIG Manager For Kubernetes
ContainerManage MIG partitions in Kubernetes with a simple label change to a node.
NVIDIA GPU Operator
ContainerDeploy and Manage NVIDIA GPU resources in Kubernetes.
Validator for NVIDIA GPU Operator
ContainerValidates NVIDIA GPU Operator components
NVIDIA Container Toolkit
ContainerRun GPU accelerated Docker containers using the NVIDIA GPU Operator
NVIDIA GPU Driver
ContainerProvision the NVIDIA GPU driver using containers.
M-Star CFD
ContainerM-Star CFD is a multi-physics modeling package used to simulate fluid flow, heat transfer, species transport, chemical reactions, particle transport, and rigid-body dynamics.
Merlin PyTorch Training
ContainerThis container allows users to do preprocessing and feature engineering with NVTabular, and then train a deep-learning based recommender system model with PyTorch.
LAMMPS
ContainerLarge-scale Atomic/Molecular Massively Parallel Simulator (LAMMPS) is a software application designed for molecular dynamics simulations.
Clara Train SDK
ContainerClara Train SDK is a domain optimized developer application framework that includes APIs for AI-Assisted Annotation, making any medical viewer AI capable and a TensorFlow based training framework with pre-trained models to kick start AI development with techniques like Transfer Learning, Federated Learning, and AutoML.
Kubevirt GPU Device Plugin
ContainerKubernetes Device Plugin built to be used for Kubevirt. Kubevirt is a Kubernetes based technology that provides a unified development platform where users can build, modify, and deploy applications residing in both Application Containers as well as VMs.
RAPIDS
ContainerThe RAPIDS suite of software libraries gives you the freedom to execute end-to-end data science and analytics pipelines entirely on GPUs.
NVIDIA Driver Manager For Kubernetes
ContainerManages NVIDIA Driver upgrades in Kubernetes cluster.
Merlin Inference
ContainerThis container allows users to deploy NVTabular workflows and HugeCTR or TensorFlow models to Triton Inference server for production.