Skip to content

Commit 8bd4e7b

Browse files
Fix NVIDIA APT GPG keys (#568)
* Fix NVIDIA APT GPG keys * Restyled by whitespace Co-authored-by: Restyled.io <commits@restyled.io>
1 parent f7623eb commit 8bd4e7b

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

task/common/machine/script.go

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,17 @@ rclone copy "$RCLONE_REMOTE/data" /tmp/tpi-task
104104
105105
yes | /etc/profile.d/install-driver-prompt.sh # for GCP GPU machines
106106
107+
# FIX NVIDIA APT GPG KEYS (https://github.com/NVIDIA/cuda-repo-management/issues/1#issuecomment-1111490201) 🤬
108+
if test -f /etc/apt/sources.list.d/cuda.list; then
109+
for list in cuda nvidia-ml; do mv /etc/apt/sources.list.d/$list.list{,.backup}; done
110+
apt-get update
111+
apt-get install --yes gpg
112+
apt-key del 7fa2af80
113+
apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/3bf863cc.pub
114+
apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64/7fa2af80.pub
115+
for list in cuda nvidia-ml; do mv /etc/apt/sources.list.d/$list.list{.backup,}; done
116+
fi
117+
107118
sudo systemctl daemon-reload
108119
sudo systemctl enable tpi-task.service --now
109120
sudo systemctl disable --now apt-daily.timer

0 commit comments

Comments
 (0)