New! DGX-1/DGX-2 Alternative: 8x Tesla V100 Server with NVLink
New! Workstation with 4x RTX 2080 Ti and Server with 8x 2080 Ti

Lambda Stack

One-line installation of TensorFlow, Keras, Caffe, Caffe, CUDA, cuDNN, and NVIDIA Drivers for Ubuntu 16.04 and18.04.

What is Lambda Stack?

Lambda Stack provides an easy way to install popular Machine Learning frameworks. With Lambda Stack, you can use apt / aptitude to install TensorFlow, Keras, PyTorch, Caffe, Caffe 2, Theano, CUDA, cuDNN, and NVIDIA GPU drivers. If a new version of any framework is released, Lambda Stack manages the upgrade. You'll never run into issues with your NVIDIA drivers again. It now supports Docker Image creation: https://github.com/lambdal/lambda-stack-dockerfiles.

Lambda Stack details

Ubuntu 16.04 / 18.04 desktop installation

For desktop computers running Ubuntu 16.04 or 18.04.

LAMBDA_REPO=$(mktemp) && \
wget -O${LAMBDA_REPO} https://lambdalabs.com/static/misc/lambda-stack-repo.deb && \
sudo dpkg -i ${LAMBDA_REPO} && rm -f ${LAMBDA_REPO} && \
sudo apt-get update && sudo apt-get install -y lambda-stack-cuda

Lambda Stack Docker image creation for use with nvidia-docker

We've created Dockerfiles with a rolling-release of Lambda Stack, designed for use with nvidia-docker. To get the Dockerfiles, clone this repository: https://github.com/lambdal/lambda-stack-dockerfiles.

After cloning, you can build an Ubuntu 18.04 Docker image with:

docker build -f Dockerfile.bionic .

For an Ubuntu 16.04 Docker image, use:

docker build -f Dockerfile.xenial .

Ubuntu 16.04 server installation

For servers running Ubuntu 16.04 without a GUI (i.e. Ubuntu 16.04 server edition).

LAMBDA_REPO=$(mktemp) && \
wget -O${LAMBDA_REPO} https://lambdalabs.com/static/misc/lambda-stack-repo.deb && \
sudo dpkg -i ${LAMBDA_REPO} && rm -f ${LAMBDA_REPO} && \
sudo apt-get update && \
sudo apt-get --yes upgrade && \
sudo apt-get install --yes --no-install-recommends lambda-server && \
sudo apt-get install --yes --no-install-recommends nvidia-390 libcuda1-390 nvidia-opencl-icd-390 && \
sudo apt-get install --yes --no-install-recommends lambda-stack-cuda

Ubuntu 18.04 server installation

For servers running Ubuntu 18.04 without a GUI (i.e. Ubuntu 18.04 server edition).

LAMBDA_REPO=$(mktemp) && \
wget -O${LAMBDA_REPO} https://lambdalabs.com/static/misc/lambda-stack-repo.deb && \
sudo dpkg -i ${LAMBDA_REPO} && rm -f ${LAMBDA_REPO} && \
sudo apt-get update && \
sudo apt-get --yes upgrade && \
sudo apt-get install --yes --no-install-recommends lambda-server && \
sudo apt-get install --yes --no-install-recommends nvidia-headless-390 nvidia-utils-390 && \
sudo apt-get install --yes --no-install-recommends lambda-stack-cuda

Incorporating into shell scripts, Docker, Ansible, etc.

If you want to integrate Lambda Stack installation into a script, you'll likely want to avoid all user input prompts. To use Lambda Stack in this way, you must have read and agreed to the CUDNN license.

LAMBDA_REPO=$(mktemp) && \
wget -O${LAMBDA_REPO} https://lambdalabs.com/static/misc/lambda-stack-repo.deb && \
sudo dpkg -i ${LAMBDA_REPO} && rm -f ${LAMBDA_REPO} && \
sudo apt-get update && \
sudo apt-get --yes upgrade && \
echo "cudnn cudnn/license_preseed select ACCEPT" | sudo debconf-set-selections && \
sudo apt-get install --yes --no-install-recommends lambda-server && \
sudo apt-get install --yes --no-install-recommends nvidia-390 libcuda1-390 nvidia-opencl-icd-390 && \
sudo apt-get install --yes --no-install-recommends lambda-stack-cuda

How to update / upgrade to the latest Lambda Stack version

Do this if a new version of TensorFlow (or any other framework) is released and you want to upgrade.

sudo apt-get update && sudo apt-get dist-upgrade && sudo apt-get autoremove

This will upgrade all packages, including dependencies such as CUDA, cuDNN, and NVIDIA drivers.

Benefits of Lambda Stack

Save weeks of setup time

You’ll be ready to start training just moments after the installation.

Make money

Calculate how many hours you'll save after you stop futzing with your Linux install and multiply that by your hourly rate!

Simple and scalable

Lambda Stack works on notebooks, workstations, servers, cloud instances, and more.

Give us feedback.

Have a question or suggestion on how we can improve Lambda Stack? Send us an email:

software@lambdalabs.com