Bump CUDA base-image

to 12.6.3-runtime-ubuntu20.04 for python 3.8 and 3.10 and to 12.6.3-runtime-ubuntu22.04 for python 3.11

Author: hc2p

.circleci/config.yml

@@ -316,16 +316,16 @@ workflows:
 
       - build-and-push-gpu:
           name: Tensorflow 2.9
-          cuda-tag: "11.4.3-runtime-ubuntu20.04"
+          cuda-tag: "12.6.3-runtime-ubuntu20.04"
           tf-version: "2.9.1"
           python-version: "3.8"
       - build-and-push-gpu:
           name: Tensorflow 2.11
-          cuda-tag: "11.4.3-runtime-ubuntu20.04"
+          cuda-tag: "12.6.3-runtime-ubuntu20.04"
           tf-version: "2.11.1"
           python-version: "3.10"
       - build-and-push-gpu:
           name: Tensorflow 2.15
-          cuda-tag: "12.2.2-runtime-ubuntu22.04"
+          cuda-tag: "12.6.3-runtime-ubuntu22.04"
           tf-version: "2.15.0"
           python-version: "3.11"

gpu/Dockerfile

@@ -1,4 +1,4 @@
-ARG CUDA_IMAGE_TAG=11.4.3-runtime-ubuntu20.04
+ARG CUDA_IMAGE_TAG=12.6.3-runtime-ubuntu20.04
 FROM nvidia/cuda:${CUDA_IMAGE_TAG}
 
 ENV DEBIAN_FRONTEND=noninteractive