File tree Expand file tree Collapse file tree 1 file changed +11
-0
lines changed Expand file tree Collapse file tree 1 file changed +11
-0
lines changed Original file line number Diff line number Diff line change @@ -104,6 +104,17 @@ rclone copy "$RCLONE_REMOTE/data" /tmp/tpi-task
104
104
105
105
yes | /etc/profile.d/install-driver-prompt.sh # for GCP GPU machines
106
106
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
+
107
118
sudo systemctl daemon-reload
108
119
sudo systemctl enable tpi-task.service --now
109
120
sudo systemctl disable --now apt-daily.timer
You can’t perform that action at this time.
0 commit comments