Lambda Stack: Deep Learning Software for Ubuntu

Lambda Stack is a curated set of software packages widely used in machine learning and deep learning. Each package is tested for compatibility and interoperability across all Lambda systems and architectures, including the latest NVIDIA HGX B200 and H200 SXM GPUs.

Supported packages

Installing Lambda Stack

Lambda Stack comes preinstalled with Ubuntu on all Lambda products, including 1-Click Clusters and Instances. It can also be installed on non-Lambda servers and workstations running Ubuntu using the Lambda Stack Install Script (licensed under BSD-3-Clause):

Non-interactive installations

For non-interactive installations, such as on servers, run:

wget -nv -O- https://lambda.ai/install-lambda-stack.sh |
I_AGREE_TO_THE_CUDNN_LICENSE=1 sh -

Note: You must read and agree to the cuDNN license agreement before running the non-interactive installation command.

Interactive installations

For interactive installations, such as on workstations, run:

wget -nv -O- https://lambda.ai/install-lambda-stack.sh | sh -
sudo reboot

Note: The installation script automatically detects your system’s hardware and installs the appropriate drivers and firmware, including NVIDIA Fabric Manager for NVSwitch-based systems.

Lambda Stack in Python virtual environments

Once Lambda Stack is installed — either preinstalled or manually — you can access its packages from within a Python virtual environment:

python3 -m venv --system-site-packages <VIRTUAL-ENVIRONMENT-NAME>

source <VIRTUAL-ENVIRONMENT-NAME>/bin/activate

Running GPU-accelerated Docker containers

Since Lambda Stack includes the NVIDIA Container Toolkit, you can run GPU-accelerated containers immediately — no additional setup required.

Check out some of our tutorials using GPU-accelerated containers:

• Using Multi-Instance GPU (MIG)
• Serving the Llama 3.1 8B and 70B models on Lambda Cloud on-demand instances
• Running DeepSeek-R1 70B using Ollama

Upgrading Lambda Stack

Run this command and all of your AI software, from PyTorch® to CUDA, will be updated. Like magic.

sudo apt update && sudo apt dist-upgrade

Current package versions

Everyone loves Lambda Stack

Used by the F500, research labs, and the Department of Defense. Everything we ship comes pre-installed with Lambda Stack. It’s loved by thousands of Lambda customers.

Ready to get started?

Create a cloud account to launch an instance today.

Launch GPU instance