fix(mlc-llm): install CUDA toolkit, use case

JJGadgets · Oct 12, 2024 · 4ca3782 · 4ca3782
1 parent 048cdf7
commit 4ca3782
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/apps/mlc-llm/Dockerfile b/apps/mlc-llm/Dockerfile
@@ -32,7 +32,7 @@ case "${CHANNEL}" in \
   cuda*) \
     apt-key del 7fa2af80 || true; \
     apt update && apt install --yes --no-install-recommends software-properties-common && apt-add-repository contrib -y && apt purge --yes software-properties-common && \
-    wget "https://developer.download.nvidia.com/compute/cuda/repos/${ID}${VERSION_ID}/$(uname -m)/cuda-keyring_1.1-1_all.deb" && apt install cuda-keyring_1.1-1_all.deb || exit 1; \
+    wget "https://developer.download.nvidia.com/compute/cuda/repos/${ID}${VERSION_ID}/$(uname -m)/cuda-keyring_1.1-1_all.deb" && apt install ./cuda-keyring_1.1-1_all.deb || exit 1; rm -rf ./cuda-keyring_1.1-1_all.deb \
     apt update && apt install --yes --no-install-recommends cuda-toolkit \
   ;; \
   *) \