From 4ca3782a016e8095021151b335e178c48fdc9b7d Mon Sep 17 00:00:00 2001
From: JJGadgets <git@jjgadgets.tech>
Date: Sun, 13 Oct 2024 06:45:22 +0800
Subject: [PATCH] fix(mlc-llm): install CUDA toolkit, use case

---
 apps/mlc-llm/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/apps/mlc-llm/Dockerfile b/apps/mlc-llm/Dockerfile
index 1ef30c8b..d20be06d 100644
--- 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 \
   ;; \
   *) \