[ci] try substituting nvcr.io/nvidia/pytorch:24.03-py3 with pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime

Dockerfile

@@ -10,7 +10,7 @@ ARG POETRY_CACHE_DIR=/tmp/poetry_cache
 ARG DEPENDENCY_GROUP=fastapi
 
 
-FROM nvcr.io/nvidia/pytorch:24.03-py3 as builder_global
+FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime as builder_global
 
 LABEL authors="[email protected]"
 
@@ -70,7 +70,7 @@ RUN git clone -n --depth=1 --filter=tree:0 https://huggingface.co/spaces/aletrn/
 WORKDIR ${LAMBDA_TASK_ROOT}
 RUN ls -l ${LAMBDA_TASK_ROOT}//*
 
-FROM nvcr.io/nvidia/pytorch:24.03-py3 as runtime
+FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime as runtime
 
 ARG ARCH
 ARG LAMBDA_TASK_ROOT