Revert cuda updates (#365)

* Revert "chore: Bump CUDA toolkit to 12.6 (#350)"

This reverts commit 6d6d7ec.

* Revert "update cudarc version (#362)"

This reverts commit 54726b5.

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

@@ -11,7 +11,7 @@ on:
 env:
   REGISTRY: ghcr.io
   IMAGE_NAME: ${{ github.repository }}-base
-  CUDA_VERSION: 12_6
+  CUDA_VERSION: 12_2
   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 --bin upgrade-checker
 
-FROM --platform=linux/amd64 ghcr.io/worldcoin/iris-mpc-base:cuda12_6-nccl2_22_3_1
+FROM --platform=linux/amd64 ghcr.io/worldcoin/gpu-iris-mpc-base:cuda12_2-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-6 libnccl2=2.22.3-1+cuda12.6 libnccl-dev=2.22.3-1+cuda12.6
+    && apt-get install -y cuda-toolkit-12-2 libnccl2=2.22.3-1+cuda12.2 libnccl-dev=2.22.3-1+cuda12.2

iris-mpc-gpu/Cargo.toml

@@ -6,7 +6,7 @@ edition = "2021"
 
 [dependencies]
 bincode = "1.3.3"
-cudarc = { version = "0.12.1", features = ["cuda-12060", "nccl"] }
+cudarc = { version = "0.12", features = ["cuda-12020", "nccl"] }
 eyre.workspace = true
 tracing.workspace = true
 bytemuck.workspace = true