diff --git a/Dockerfile b/Dockerfile index 899d34d..8aa32bb 100644 --- a/Dockerfile +++ b/Dockerfile @@ -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