From 63acbdb34aa9c1050b3ca5484f620dcd323c6ed1 Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Thu, 8 Aug 2024 18:06:01 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B8=85=E5=8D=8E=E6=BA=90=E7=9A=84pyyaml?= =?UTF-8?q?=E7=9A=846.0.2=E7=89=88=E6=9C=AC=E6=9A=82=E6=97=B6=E6=97=A0?= =?UTF-8?q?=E6=B3=95=E4=B8=8B=E8=BD=BD=EF=BC=8C=E6=89=8B=E5=8A=A8=E8=AE=BE?= =?UTF-8?q?=E4=B8=BA6.0.1?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- Dockerfile | 4 ++-- requirements.txt | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 8818a68..46ab57f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -9,14 +9,14 @@ ENV PYTHONUNBUFFERED=1 \ # 设置时区 TZ=Asia/Shanghai \ # 设置pip镜像地址,加快安装速度 - PIP_INDEX_URL=http://mirrors.aliyun.com/pypi/simple + PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple # 将当前目录内容复制到容器的/app内 COPY . /app # 安装任何需要的包 RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \ - && pip install --no-cache-dir --trusted-host mirrors.aliyun.com -r requirements.txt \ + && 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/ diff --git a/requirements.txt b/requirements.txt index f408f85..14e0c0f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,5 @@ +pyyaml==6.0.1 + numpy==1.26.4 onnxconverter-common==1.14.0 opencv-python==4.6.0.66