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:
Lambda Stack details
Ubuntu 16.04 / 18.04 desktop installation
For desktop computers running Ubuntu 16.04 or 18.04.
after this, you'll need to run sudo reboot to complete the installation.
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:
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.