diff --git a/slurm-packages/Dockerfile b/slurm-packages/Dockerfile index 6e8241a..750ba65 100644 --- a/slurm-packages/Dockerfile +++ b/slurm-packages/Dockerfile @@ -1,4 +1,4 @@ -ARG BASE_IMAGE=cr.eu-north1.nebius.cloud/soperator/cuda_base:12.9.0-ubuntu24.04-nccl2.26.5-1-295cb71 +ARG BASE_IMAGE=cr.eu-north1.nebius.cloud/soperator/cuda:12.9.0-cudnn-devel-ubuntu22.04 FROM $BASE_IMAGE