NVIDIA B200s are live on Lambda Cloud! Set up your Demo today! 
Login Create account
lambda_square_stack-1

Lambda Stack is all the AI software you need

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 include:

TensorFlow_logo-color Pytorch_logo-color Keras_logo-color Ubuntu_logo-color NVIDIA_Cuda_logo-color

Installing Lambda Stack

Lambda Stack comes preinstalled with Ubuntu on all Lambda products, including 1-Click Clusters, On-Demand Cloud instances, and the Vector line of workstations and desktops. It can also be installed on non-Lambda servers and workstations running Ubuntu with a single command.
Icon - Non-interactive installations

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.

Icon - Interactive installations

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.

Using 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:

lambda-stack

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
Heart

Everyone loves Lambda Stack—used by the F500, research labs, and the DOD

Every laptop, workstation, and server that we ship comes pre-installed with Lambda Stack. It's loved by thousands of Lambda customers.
Intuitive_logo writer_logo sony_logo samsung_logo pika_logo

Ready to get started?

Create a cloud account instantly to spin up GPUs today or contact us to secure a long-term contract for thousands of GPUs