chore: Bump CUDA toolkit to 12.6 (#350)

* chore: Bump CUDA toolkit to 12.6

* fixed name of the repo

Author: wojciechsromek

.github/workflows/build-and-push-base.yaml

@@ -11,7 +11,7 @@ on:
 env:
   REGISTRY: ghcr.io
   IMAGE_NAME: ${{ github.repository }}-base
-  CUDA_VERSION: 12_2
+  CUDA_VERSION: 12_6
   NCCL_VERSION: 2_22_3_1
 jobs:
   docker:

Dockerfile

@@ -29,7 +29,7 @@ WORKDIR /src/gpu-iris-mpc
 COPY . .
 RUN cargo build --release --target x86_64-unknown-linux-gnu --bin server --bin client --bin key-manager --bin upgrade-server --bin upgrade-client
 
-FROM --platform=linux/amd64 ghcr.io/worldcoin/gpu-iris-mpc-base:cuda12_2-nccl2_22_3_1
+FROM --platform=linux/amd64 ghcr.io/worldcoin/iris-mpc-base:cuda12_6-nccl2_22_3_1
 ENV DEBIAN_FRONTEND=noninteractive
 
 # Include client, server and key-manager, upgrade-client and upgrade-server binaries

Dockerfile.base

@@ -5,4 +5,4 @@ RUN apt-get update && apt-get install -y pkg-config wget libssl-dev ca-certifica
 RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
     && dpkg -i cuda-keyring_1.1-1_all.deb \
     && apt-get update \
-    && apt-get install -y cuda-toolkit-12-2 libnccl2=2.22.3-1+cuda12.2 libnccl-dev=2.22.3-1+cuda12.2
+    && apt-get install -y cuda-toolkit-12-6 libnccl2=2.22.3-1+cuda12.6 libnccl-dev=2.22.3-1+cuda12.6