diff --git a/serverless/pytorch/foolwood/siammask/nuclio/function-gpu.yaml b/serverless/pytorch/foolwood/siammask/nuclio/function-gpu.yaml
index 119c3f54..154c1b78 100644
--- a/serverless/pytorch/foolwood/siammask/nuclio/function-gpu.yaml
+++ b/serverless/pytorch/foolwood/siammask/nuclio/function-gpu.yaml
@@ -18,7 +18,7 @@ spec:
 
   build:
     image: cvat.pth.foolwood.siammask
-    baseImage: nvidia/cuda:11.1-devel-ubuntu20.04
+    baseImage: nvidia/cuda:11.1.1-devel-ubuntu20.04
 
     directives:
       preCopy: