From b6126146dca1cba1f4464bd37aba74c309562c25 Mon Sep 17 00:00:00 2001
From: "mulin.lyh" <mulin.lyh@taobao.com>
Date: Tue, 5 Dec 2023 20:39:46 +0800
Subject: [PATCH] add cuda compat to LD_LIBRARY_PATH

---
 docker/Dockerfile.ubuntu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu
index 93308e25..020eb550 100644
--- a/docker/Dockerfile.ubuntu
+++ b/docker/Dockerfile.ubuntu
@@ -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