From dbc2ec59b5603fbd25f3833b2407b4f3612b9f02 Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Fri, 14 Feb 2025 14:48:40 +0200 Subject: [PATCH] docker : drop to CUDA 12.4 (#11869) * docker : drop to CUDA 12.4 * docker : update readme [no ci] --- .devops/cuda.Dockerfile | 2 +- docs/docker.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.devops/cuda.Dockerfile b/.devops/cuda.Dockerfile index 974dd78a8..0bc4e7ee1 100644 --- a/.devops/cuda.Dockerfile +++ b/.devops/cuda.Dockerfile @@ -1,6 +1,6 @@ ARG UBUNTU_VERSION=22.04 # This needs to generally match the container host's environment. -ARG CUDA_VERSION=12.6.0 +ARG CUDA_VERSION=12.4.0 # Target the CUDA build image ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION} diff --git a/docs/docker.md b/docs/docker.md index cab5ae957..58b5d381d 100644 --- a/docs/docker.md +++ b/docs/docker.md @@ -69,7 +69,7 @@ You may want to pass in some different `ARGS`, depending on the CUDA environment The defaults are: -- `CUDA_VERSION` set to `12.6.0` +- `CUDA_VERSION` set to `12.4.0` - `CUDA_DOCKER_ARCH` set to the cmake build default, which includes all the supported architectures The resulting images, are essentially the same as the non-CUDA images: