We support up to CUDA 10.0 on Ubuntu 16.04 & 18.04. We support all the way up to CUDA 10.2 on Ubuntu 20.04.
Run the following to upgrade to 20.04:
sudo sed -i '/^AutomaticLoginEnable/ s/true/false/' /etc/gdm3/custom.conf &&\
sudo apt -y update && sudo apt -y dist-upgrade && do-release-upgrade -d -f DistUpgradeViewNonInteractive
Once the update completes, reboot & reinstall Lambda Stack:
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