Use CUDA 11.8 container images, 12.0 for head

Signed-off-by: Enrico Minack <[email protected]>
horovod · Mar 27, 2024 · 34c4a4c · 34c4a4c
1 parent 66d3c02
commit 34c4a4c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/docker-compose.test.yml b/docker-compose.test.yml
@@ -175,7 +175,7 @@ services:
  extends: test-gpu-base
  build:
  args:
- CUDA_DOCKER_VERSION: 11.6.2-devel-ubuntu20.04
+ CUDA_DOCKER_VERSION: 11.8.0-devel-ubuntu20.04
  CUDNN_VERSION: 8.4.1.50-1+cuda11.6
  NCCL_VERSION_OVERRIDE: 2.11.4-1+cuda11.6
  PYTHON_VERSION: 3.8
@@ -190,7 +190,7 @@ services:
  extends: test-gpu-base
  build:
  args:
- CUDA_DOCKER_VERSION: 11.6.2-devel-ubuntu20.04
+ CUDA_DOCKER_VERSION: 11.8.0-devel-ubuntu20.04
  CUDNN_VERSION: 8.4.1.50-1+cuda11.6
  NCCL_VERSION_OVERRIDE: 2.11.4-1+cuda11.6
  TENSORFLOW_PACKAGE: tensorflow-gpu==2.10.1
@@ -203,7 +203,7 @@ services:
  extends: test-gpu-base
  build:
  args:
- CUDA_DOCKER_VERSION: 11.6.2-devel-ubuntu20.04
+ CUDA_DOCKER_VERSION: 11.8.0-devel-ubuntu20.04
  CUDNN_VERSION: 8.4.1.50-1+cuda11.6
  NCCL_VERSION_OVERRIDE: 2.11.4-1+cuda11.6
  # tensorflow package supports GPU from 2.11.1 and 2.12.0 on
@@ -232,7 +232,7 @@ services:
  extends: test-gpu-base
  build:
  args:
- CUDA_DOCKER_VERSION: 11.8.0-devel-ubuntu20.04
+ CUDA_DOCKER_VERSION: 12.0.0-devel-ubuntu20.04
  CUDNN_VERSION: 8.6.0.163-1+cuda11.8
  NCCL_VERSION_OVERRIDE: 2.16.5-1+cuda11.8
  MPI_KIND: OpenMPI
@@ -248,7 +248,7 @@ services:
  extends: test-gpu-base
  build:
  args:
- CUDA_DOCKER_VERSION: 11.6.2-devel-ubuntu20.04
+ CUDA_DOCKER_VERSION: 11.8.0-devel-ubuntu20.04
  CUDNN_VERSION: 8.4.1.50-1+cuda11.6
  NCCL_VERSION_OVERRIDE: 2.11.4-1+cuda11.6
  MPI_KIND: OpenMPI