diff --git a/nnunet_for_pytorch.def b/nnunet_for_pytorch.def
index 31ed6290080801bd0f3994e4501ed035e8c05b39..1c015b62a89d4e15a804737e7d1aa21ff68b925d 100644
--- a/nnunet_for_pytorch.def
+++ b/nnunet_for_pytorch.def
@@ -1,5 +1,5 @@
 Bootstrap: localimage
-From: /proj/nsc_testing/xuan/containers/cuda_11.1.1-cudnn8-runtime-ubuntu18.04.sif
+From: /proj/nsc_testing/xuan/containers/nvidia_pytorch:21.11-py3.sif
 
 
 %environment