Skip to content

Commit 47ccfcc

Browse files
authored
fix(ci): add nvidia-l4t capability to l4t images (#5914)
Signed-off-by: Ettore Di Giacinto <[email protected]>
1 parent a760f7f commit 47ccfcc

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

Dockerfile

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,12 @@ RUN <<EOT bash
7272
fi
7373
EOT
7474

75+
RUN <<EOT bash
76+
if [ "${BUILD_TYPE}" = "cublas" ] && [ "${TARGETARCH}" = "arm64" ]; then
77+
echo "nvidia-l4t" > /run/localai/capability
78+
fi
79+
EOT
80+
7581
# If we are building with clblas support, we need the libraries for the builds
7682
RUN if [ "${BUILD_TYPE}" = "clblas" ] && [ "${SKIP_DRIVERS}" = "false" ]; then \
7783
apt-get update && \

0 commit comments

Comments
 (0)