add cuda compat to LD_LIBRARY_PATH

This commit is contained in:
mulin.lyh
2023-12-05 20:39:46 +08:00
parent 442b45f0c3
commit b6126146dc

View File

@@ -56,3 +56,4 @@ RUN pip install --no-cache-dir --upgrade pip && \
COPY examples /modelscope/examples
ENV SETUPTOOLS_USE_DISTUTILS=stdlib
ENV VLLM_USE_MODELSCOPE=True
ENV LD_LIBRARY_PATH=/usr/local/cuda/compat:$LD_LIBRARY_PATH