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.
Lambda Stack is both a system wide package, a Dockerfile, and a Docker image.
No need to compromise. Lambda Stack is not only a system wide installation of all of your favorite frameworks and drivers but also a convenient "everything included" deep learning Docker image. Now you'll have your team up and running with GPU-accelerated Docker images in minutes instead of weeks. To learn more about how to set up Lambda Stack GPU Dockerfiles check out our tutorial:
Ubuntu 18.04 / Ubuntu 16.04 desktop installation
Paste this command into a fresh Ubuntu installation to install Lambda stack on your desktop system. For servers, see the server installation section.
after this, you'll need to run sudo reboot to complete the installation.
Lambda Stack details
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. You can either follow our Lambda Stack GPU Docker tutorial here. Or, if you already have nvidia-docker installed you can build the Docker image immediately:
Using Lambda Stack with python virtual environments
We're often asked how to best use Lambda Stack with a python virtual environment. You have two choices: use Lambda Stack as a way to install CUDA, CuDNN, and NVIDIA drivers; or, use Lambda Stack as a way to manage TensorFlow and PyTorch as well as CUDA, CuDNN, NVIDIA drivers. Here's how to do that:
Here's how to do it where the TensorFlow version is managed within the virtual environment:
Ubuntu 16.04 server installation
For servers running Ubuntu 16.04 without a GUI (i.e. Ubuntu 16.04 server edition).
Ubuntu 18.04 server installation
For servers running Ubuntu 18.04 without a GUI (i.e. Ubuntu 18.04 server edition).
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.
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.
This will upgrade all packages, including dependencies such as CUDA, cuDNN, and NVIDIA drivers.
You’ll be ready to start training just moments after the installation.
Calculate how many hours you'll save after you stop futzing with your Linux install and multiply that by your hourly rate!
Lambda Stack works on notebooks, workstations, servers, cloud instances, and more.