调整docker中依赖的安装

This commit is contained in:
2024-07-18 15:24:20 +08:00
parent cb60802d24
commit 9bd7038d5f
3 changed files with 6 additions and 2 deletions

View File

@@ -15,7 +15,9 @@ ENV VERSION=${VERSION} \
COPY . /app
# 安装任何需要的包
RUN pip install --no-cache-dir -r requirements.txt
RUN pip install --no-cache-dir -r requirements.txt \
&& pip uninstall -y onnxruntime onnxruntime-gpu \
&& pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
# 暴露端口
# EXPOSE 8081