From 3431ab332c82b67521fb03c506988cbc6bb6e437 Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Thu, 12 Oct 2023 07:16:58 +0000
Subject: [PATCH] Update file U-Net3D.def

---
 MLPerf/training/image_segmentation/pytorch/U-Net3D.def | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/MLPerf/training/image_segmentation/pytorch/U-Net3D.def b/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
index 59ba215..3496f72 100644
--- a/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
+++ b/MLPerf/training/image_segmentation/pytorch/U-Net3D.def
@@ -1,5 +1,5 @@
 Bootstrap: localimage
-From: /proj/nsc_testing/xuan/containers/pytorch_1.7.1-cuda11.0-cudnn8-devel.sif
+From: /proj/nsc_testing/xuan/containers/pytorch_1.7.1-cuda11.0-cudnn8-runtime.sif
 
 
 %environment
-- 
GitLab