This commit is contained in:
hailin 2025-07-02 15:28:40 +08:00
parent 5e4563b87e
commit b4c31eb5ce
1 changed files with 5 additions and 2 deletions

View File

@ -89,9 +89,12 @@ 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 动态库(关键
# 👉 拷贝 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 /usr/local/cuda/lib64/libcupti.so /usr/lib/x86_64-linux-gnu/
# 👇建议在后面补上
RUN ldconfig
COPY --from=builder-extras /wheels /tmp/wheels
RUN python3 -m pip install --no-cache-dir /tmp/wheels/* && rm -rf /tmp/wheels