Skip to content

Commit b5efc4f

Browse files
authored
chore(cudss): add cudds to l4t images (#6338)
* chore(cudds): add cudds to l4t images Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * add arm64 to CI tests Signed-off-by: Ettore Di Giacinto <mudler@localai.io> --------- Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
1 parent 3f9c09a commit b5efc4f

File tree

2 files changed

+20
-0
lines changed

2 files changed

+20
-0
lines changed

.github/workflows/image-pr.yml

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,3 +66,13 @@ jobs:
6666
runs-on: 'ubuntu-latest'
6767
base-image: "ubuntu:22.04"
6868
makeflags: "--jobs=4 --output-sync=target"
69+
- build-type: 'cublas'
70+
cuda-major-version: "12"
71+
cuda-minor-version: "0"
72+
platforms: 'linux/arm64'
73+
tag-latest: 'false'
74+
tag-suffix: '-nvidia-l4t-arm64'
75+
base-image: "nvcr.io/nvidia/l4t-jetpack:r36.4.0"
76+
runs-on: 'ubuntu-24.04-arm'
77+
makeflags: "--jobs=4 --output-sync=target"
78+
skip-drivers: 'true'

Dockerfile

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,16 @@ RUN <<EOT bash
7878
fi
7979
EOT
8080

81+
# https://github.com/NVIDIA/Isaac-GR00T/issues/343
82+
RUN <<EOT bash
83+
if [ "${BUILD_TYPE}" = "cublas" ] && [ "${TARGETARCH}" = "arm64" ]; then
84+
wget https://developer.download.nvidia.com/compute/cudss/0.6.0/local_installers/cudss-local-tegra-repo-ubuntu2204-0.6.0_0.6.0-1_arm64.deb && \
85+
dpkg -i cudss-local-tegra-repo-ubuntu2204-0.6.0_0.6.0-1_arm64.deb && \
86+
cp /var/cudss-local-tegra-repo-ubuntu2204-0.6.0/cudss-*-keyring.gpg /usr/share/keyrings/ && \
87+
apt-get update && apt-get -y install cudss
88+
fi
89+
EOT
90+
8191
# If we are building with clblas support, we need the libraries for the builds
8292
RUN if [ "${BUILD_TYPE}" = "clblas" ] && [ "${SKIP_DRIVERS}" = "false" ]; then \
8393
apt-get update && \

0 commit comments

Comments
 (0)