File tree Expand file tree Collapse file tree 1 file changed +21
-9
lines changed Expand file tree Collapse file tree 1 file changed +21
-9
lines changed Original file line number Diff line number Diff line change @@ -32,15 +32,27 @@ RUN <<EOT bash
3232 if [ "${BUILD_TYPE}" = "vulkan" ] && [ "${SKIP_DRIVERS}" = "false" ]; then
3333 apt-get update && \
3434 apt-get install -y --no-install-recommends \
35- software-properties-common pciutils wget gpg-agent && \
36- wget -qO - https://packages.lunarg.com/lunarg-signing-key-pub.asc | apt-key add - && \
37- wget -qO /etc/apt/sources.list.d/lunarg-vulkan-jammy.list https://packages.lunarg.com/vulkan/lunarg-vulkan-jammy.list && \
38- apt-get update && \
39- apt-get install -y \
40- vulkan-sdk && \
41- apt-get clean && \
42- rm -rf /var/lib/apt/lists/* && \
43- echo "vulkan" > /run/localai/capability
35+ software-properties-common pciutils wget gpg-agent curl xz-utils && \
36+ echo "vulkan" > /run/localai/capability && \
37+ if [ "amd64" = "$TARGETARCH" ]; then
38+ wget -qO - https://packages.lunarg.com/lunarg-signing-key-pub.asc | apt-key add - && \
39+ wget -qO /etc/apt/sources.list.d/lunarg-vulkan-jammy.list https://packages.lunarg.com/vulkan/lunarg-vulkan-jammy.list && \
40+ apt-get update && \
41+ apt-get install -y \
42+ vulkan-sdk && \
43+ apt-get clean && \
44+ rm -rf /var/lib/apt/lists/*
45+ fi
46+ if [ "arm64" = "$TARGETARCH" ]; then
47+ # For ARM64, we need to build the Vulkan SDK manually as there are no packages available
48+ mkdir vulkan && cd vulkan && curl -o vulkan-sdk.tar.xz https://sdk.lunarg.com/sdk/download/latest/linux/vulkan-sdk.tar.xz && \
49+ tar -xJf vulkan-sdk.tar.xz && \
50+ rm vulkan-sdk.tar.xz && \
51+ cd * && \
52+ ./vulkansdk -j 1 && \
53+ cd ../.. && \
54+ rm -rf vulkan
55+ fi
4456 fi
4557EOT
4658
You can’t perform that action at this time.
0 commit comments