diff --git a/.github/workflows/build-and-push-base.yaml b/.github/workflows/build-and-push-base.yaml index e35c72b60..f9b8ce4b7 100644 --- a/.github/workflows/build-and-push-base.yaml +++ b/.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: diff --git a/Dockerfile b/Dockerfile index f34fbac2a..f3f6f134d 100644 --- a/Dockerfile +++ b/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/gpu-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 diff --git a/Dockerfile.base b/Dockerfile.base index 782f40900..b77f852a8 100644 --- a/Dockerfile.base +++ b/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