From 5e4563b87e7b720d6e7bcc2ecc472afe40e48e85 Mon Sep 17 00:00:00 2001
From: hailin <hailin@gdzx.xyz>
Date: Wed, 2 Jul 2025 15:26:24 +0800
Subject: [PATCH] .

---
 Dockerfile | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 5a1b43b..899d34d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -89,6 +89,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     rm -rf /var/lib/apt/lists/* && \
     python3 -m pip install --no-cache-dir --upgrade pip
 
+# 👉 拷贝 cupti 动态库（关键）
+COPY --from=builder-extras /usr/local/cuda/lib64/libcupti.so.12 /usr/lib/x86_64-linux-gnu/
+COPY --from=builder-extras /usr/local/cuda/lib64/libcupti.so.12.235.208 /usr/lib/x86_64-linux-gnu/    
+
 COPY --from=builder-extras /wheels /tmp/wheels
 RUN python3 -m pip install --no-cache-dir /tmp/wheels/* && rm -rf /tmp/wheels