This commit is contained in:
parent
a447daf3d8
commit
5e4563b87e
|
|
@ -89,6 +89,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||||
rm -rf /var/lib/apt/lists/* && \
|
rm -rf /var/lib/apt/lists/* && \
|
||||||
python3 -m pip install --no-cache-dir --upgrade pip
|
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
|
COPY --from=builder-extras /wheels /tmp/wheels
|
||||||
RUN python3 -m pip install --no-cache-dir /tmp/wheels/* && rm -rf /tmp/wheels
|
RUN python3 -m pip install --no-cache-dir /tmp/wheels/* && rm -rf /tmp/wheels
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue