fix(nccl-tests): use cuda-nvcc-12-6 to support Ampere/Volta (sm_70..sm_90)

This commit is contained in:
2026-03-26 23:50:36 +03:00
parent 5407c26e25
commit 896bdb6ee8
3 changed files with 9 additions and 5 deletions

View File

@@ -36,7 +36,7 @@ RUN wget -qO /tmp/cuda-keyring.gpg \
https://developer.download.nvidia.com/compute/cuda/repos/debian12/x86_64/ /" \
> /etc/apt/sources.list.d/cuda.list \
&& apt-get update -qq \
&& apt-get install -y cuda-nvcc-13-0 \
&& apt-get install -y cuda-nvcc-12-6 \
&& rm -rf /var/lib/apt/lists/*
RUN arch="$(dpkg --print-architecture)" \