Fast setup for every time you spinup a lambda instance

Lamda-Update.sh

sudo apt-add-repository -r ppa:graphics-drivers/ppa
sudo apt update
sudo apt remove nvidia*
sudo apt autoremove
sudo apt-get install ubuntu-drivers-common
sudo ubuntu-drivers autoinstall
# sudo apt -f install

sudo apt install aptitude
sudo aptitude install nvidia-440
sudo apt install nvidia-cuda-toolkit
# sudo add-apt-repository ppa:graphics-drivers/ppa
# sudo apt-get update

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-440 libcuda1-440 nvidia-opencl-icd-440 && \
sudo apt-get install --yes --no-install-recommends lambda-stack-cuda

sudo apt-get install unzip
pip install torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html