File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -106,7 +106,7 @@ RUN <<EOT bash
106106 libcublas-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
107107 libcusparse-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
108108 libcusolver-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION}
109- if [ "arm64" = "$TARGETARCH" ]; then
109+ if [ "${CUDA_MAJOR_VERSION}" = "13" ] && [ " arm64" = "$TARGETARCH" ]; then
110110 apt-get install -y --no-install-recommends \
111111 libcufile-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} libcudnn9-cuda-${CUDA_MAJOR_VERSION} cuda-cupti-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} libnvjitlink-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION}
112112 fi
Original file line number Diff line number Diff line change @@ -94,7 +94,11 @@ RUN <<EOT bash
9494 curl -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/x86_64/cuda-keyring_1.1-1_all.deb
9595 fi
9696 if [ "arm64" = "$TARGETARCH" ]; then
97- curl -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/sbsa/cuda-keyring_1.1-1_all.deb
97+ if [ "${CUDA_MAJOR_VERSION}" = "13" ]; then
98+ curl -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/sbsa/cuda-keyring_1.1-1_all.deb
99+ else
100+ curl -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/arm64/cuda-keyring_1.1-1_all.deb
101+ fi
98102 fi
99103 dpkg -i cuda-keyring_1.1-1_all.deb && \
100104 rm -f cuda-keyring_1.1-1_all.deb && \
@@ -106,7 +110,7 @@ RUN <<EOT bash
106110 libcublas-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
107111 libcusparse-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
108112 libcusolver-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION}
109- if [ "arm64" = "$TARGETARCH" ]; then
113+ if [ "${CUDA_MAJOR_VERSION}" = "13" ] && [ " arm64" = "$TARGETARCH" ]; then
110114 apt-get install -y --no-install-recommends \
111115 libcufile-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} libcudnn9-cuda-${CUDA_MAJOR_VERSION} cuda-cupti-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} libnvjitlink-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION}
112116 fi
Original file line number Diff line number Diff line change @@ -148,11 +148,14 @@ RUN <<EOT bash
148148 apt-get install -y --no-install-recommends \
149149 software-properties-common pciutils
150150 if [ "amd64" = "$TARGETARCH" ]; then
151- echo https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/x86_64/cuda-keyring_1.1-1_all.deb
152151 curl -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/x86_64/cuda-keyring_1.1-1_all.deb
153152 fi
154153 if [ "arm64" = "$TARGETARCH" ]; then
155- curl -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/sbsa/cuda-keyring_1.1-1_all.deb
154+ if [ "${CUDA_MAJOR_VERSION}" = "13" ]; then
155+ curl -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/sbsa/cuda-keyring_1.1-1_all.deb
156+ else
157+ curl -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/arm64/cuda-keyring_1.1-1_all.deb
158+ fi
156159 fi
157160 dpkg -i cuda-keyring_1.1-1_all.deb && \
158161 rm -f cuda-keyring_1.1-1_all.deb && \
@@ -164,7 +167,7 @@ RUN <<EOT bash
164167 libcublas-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
165168 libcusparse-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
166169 libcusolver-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION}
167- if [ "arm64" = "$TARGETARCH" ]; then
170+ if [ "${CUDA_MAJOR_VERSION}" = "13" ] && [ " arm64" = "$TARGETARCH" ]; then
168171 apt-get install -y --no-install-recommends \
169172 libcufile-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} libcudnn9-cuda-${CUDA_MAJOR_VERSION} cuda-cupti-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} libnvjitlink-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION}
170173 fi
Original file line number Diff line number Diff line change @@ -108,7 +108,11 @@ RUN <<EOT bash
108108 curl -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/x86_64/cuda-keyring_1.1-1_all.deb
109109 fi
110110 if [ "arm64" = "$TARGETARCH" ]; then
111- curl -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/sbsa/cuda-keyring_1.1-1_all.deb
111+ if [ "${CUDA_MAJOR_VERSION}" = "13" ]; then
112+ curl -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/sbsa/cuda-keyring_1.1-1_all.deb
113+ else
114+ curl -O https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${UBUNTU_VERSION}/arm64/cuda-keyring_1.1-1_all.deb
115+ fi
112116 fi
113117 dpkg -i cuda-keyring_1.1-1_all.deb && \
114118 rm -f cuda-keyring_1.1-1_all.deb && \
@@ -120,7 +124,7 @@ RUN <<EOT bash
120124 libcublas-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
121125 libcusparse-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} \
122126 libcusolver-dev-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION}
123- if [ "arm64" = "$TARGETARCH" ]; then
127+ if [ "${CUDA_MAJOR_VERSION}" = "13" ] && [ " arm64" = "$TARGETARCH" ]; then
124128 apt-get install -y --no-install-recommends \
125129 libcufile-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} libcudnn9-cuda-${CUDA_MAJOR_VERSION} cuda-cupti-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION} libnvjitlink-${CUDA_MAJOR_VERSION}-${CUDA_MINOR_VERSION}
126130 fi
You can’t perform that action at this time.
0 commit comments