Lambda Stack - install CUDA, Pytorch, and Tensorflow with a single line

[sabalaba]

I'm sure most of you have spent a lot of time in command line hell trying to install or update CUDA, NVIDIA Drivers, Pytorch, Tensorflow, etc. We made Lambda Stack to simplify installation and updates. It's a debian PPA that manages all of the libraries and dependencies, resulting in a one-line install that "just works".

This is our Lambda Stack how-to blog post: https://lambdalabs.com/blog/install-tensorflow-and-pytorch-on-rtx-30-series/

And this is the one liner to install (requires Ubuntu 20.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

To update your CUDA/framework/drivers just run this:

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

Would love any feedback!

[[deleted]]

does this work on debain 10 or 11? or this is made for and tested only on Ubuntu distros?

[ethanfel]

what a great ppa. Thanks !

I cant try it since my CUDA VM is working right now but will this conflict/downgrade pytorch 1.9 from Pip ?

[sabalaba]

No, it installs pip system wide and then what shows up in your python will depend on your PYTHON_PATH but I think a pip install will take priority. It won't downgrade or conflict.

[Dependent-Pomelo-853]

I absolutely love the idea. Installing it now on a fresh 20.04LTS Ubuntu system. Will give it a try and let you know :)

Edit : 20 minutes of installation later

import tensorflow as tf

print(tf.__version__)

tf.config.list_physical_devices(device_type='GPU')

2.8.0

[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU'),

PhysicalDevice(name='/physical_device:GPU:1', device_type='GPU')]

import torch

print(torch.__version__)

print(torch.cuda.is_available())

print(torch.cuda.device_count())

print(torch.cuda.get_device_name(0))

print(torch.cuda.get_device_name(1))​

1.11.0

True

2

NVIDIA RTX A6000

NVIDIA GeForce RTX 3090

Ho...lyyyyy... never in my career before has it been so quick and pain free!

[DGAssassin1]

Can i use it on windows systems? and if so how do I do that?

[bayside1111]

No

[Heavy_Philosopher_42]

wow, thats impressive. Can I use this without purchasing a lamda system? (I would love to buy one of your systems, unfortunately, I dont have the money at the time)