This commit is contained in:
parent
7ac38fec98
commit
21e666c47c
|
|
@ -120,7 +120,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends gcc g++ build-e
|
||||||
python3 python3-dev python3-pip python3-distutils ca-certificates \
|
python3 python3-dev python3-pip python3-distutils ca-certificates \
|
||||||
libopenblas-dev libgomp1 libcupti-dev libnuma1 libopenmpi-dev openmpi-bin libnuma-dev libpng16-16 libjpeg8 && \
|
libopenblas-dev libgomp1 libcupti-dev libnuma1 libopenmpi-dev openmpi-bin libnuma-dev libpng16-16 libjpeg8 && \
|
||||||
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 \
|
||||||
|
&& python3 -m pip install --no-cache-dir xgrammar
|
||||||
|
|
||||||
# 👉 拷贝 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 /usr/lib/x86_64-linux-gnu/
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue