diff --git a/cloud/google/common.py b/cloud/google/common.py index 818a4bc..f2eaa6b 100644 --- a/cloud/google/common.py +++ b/cloud/google/common.py @@ -20,8 +20,10 @@ echo ##### Set up NVidia ############################################################# # Add the package repositories distribution=$(. /etc/os-release;echo $ID$VERSION_ID) -curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add - -curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list +curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg +curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \ +sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \ +sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list add-apt-repository -y ppa:graphics-drivers/ppa apt-get update -y DEBIAN_FRONTEND=noninteractive apt-get -y -o Dpkg::Options::="--force-confdef" -o Dpkg::Options::="--force-confold" install nvidia-headless-560 nvidia-utils-560 nvidia-container-toolkit