BUG: fix libcuda.so.1: cannot open shared object file (#883)

xorbitsai · Jan 11, 2024 · 1ba2ef7 · 1ba2ef7
1 parent 95bdf3b
commit 1ba2ef7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/xinference/deploy/docker/Dockerfile b/xinference/deploy/docker/Dockerfile
@@ -16,6 +16,7 @@ RUN apt-get -y update \
   && apt-get -yq clean
 
 ENV PATH $NVM_DIR/versions/node/v$NODE_VERSION/bin:$PATH
+ENV LD_LIBRARY_PATH /usr/local/cuda/compat:$LD_LIBRARY_PATH
 
 ARG PIP_INDEX=https://pypi.org/simple
 RUN python -m pip install --upgrade -i "$PIP_INDEX" pip && \