diff --git a/docker/cuda/Dockerfile b/docker/cuda/Dockerfile index ec49755..876fac4 100644 --- a/docker/cuda/Dockerfile +++ b/docker/cuda/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:12.1.0-base-ubuntu22.04 as builder +FROM nvidia/cuda:12.2.0-base-ubuntu22.04 as builder RUN apt-get update \ && apt-get install -y --no-install-recommends \ @@ -21,7 +21,7 @@ COPY setup.cfg \ RUN pip wheel --no-cache-dir --no-deps \ --wheel-dir /app/wheel /app -FROM nvidia/cuda:12.1.0-base-ubuntu22.04 +FROM nvidia/cuda:12.2.0-base-ubuntu22.04 RUN apt-get update \ && apt-get install -y --no-install-recommends \