From 4145132d1e60c4925186bd631f1408c27f6c1020 Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Wed, 11 Oct 2023 21:26:22 +0000
Subject: [PATCH] Update file U-Net3D.def
---
MLPerf/training/image_segmentation/pytorch/U-Net3D.def | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/MLPerf/training/image_segmentation/pytorch/U-Net3D.def b/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
index e60aa38..59ba215 100644
--- a/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
+++ b/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
@@ -12,6 +12,10 @@ export PYTHONNOUSERSITE=True
mkdir -p /workspace/unet3d
cd /workspace/unet3d
+apt-key del 7fa2af80
+apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub
+apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu2004/x86_64/7fa2af80.pub
+
apt-get update && \
apt-get upgrade -y && \
apt-get install -y git
--
GitLab