Skip to content

Commit

Permalink
Add default LD_LIBRARY_PATH env var
Browse files Browse the repository at this point in the history
  • Loading branch information
Adrian Gonzalez-Martin committed May 2, 2023
1 parent bbaa42d commit 5172beb
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -23,11 +23,16 @@ ARG CONDA_VERSION=22.11.1
ARG MINIFORGE_VERSION=${CONDA_VERSION}-4
ARG RUNTIMES="all"

# Set a few default environment variables, including `LD_LIBRARY_PATH`
# (required to use GKE's injected CUDA libraries).
# NOTE: When updating between major Python versions make sure you update the
# `/opt/conda` path within `LD_LIBRARY_PATH`.
ENV MLSERVER_MODELS_DIR=/mnt/models \
MLSERVER_ENV_TARBALL=/mnt/models/environment.tar.gz \
MLSERVER_PATH=/opt/mlserver \
CONDA_PATH=/opt/conda \
PATH=/opt/mlserver/.local/bin:/opt/conda/bin:$PATH
PATH=/opt/mlserver/.local/bin:/opt/conda/bin:$PATH \
LD_LIBRARY_PATH=/usr/local/nvidia/lib64:/opt/conda/lib/python3.8/site-packages/nvidia/cuda_runtime/lib:$LD_LIBRARY_PATH

# Install some base dependencies required for some libraries
RUN microdnf update -y && \
Expand Down

0 comments on commit 5172beb

Please sign in to comment.