清华源的pyyaml的6.0.2版本暂时无法下载,手动设为6.0.1

This commit is contained in:
2024-08-08 18:06:01 +08:00
parent c7161b0aac
commit 63acbdb34a
2 changed files with 4 additions and 2 deletions

View File

@@ -9,14 +9,14 @@ ENV PYTHONUNBUFFERED=1 \
# 设置时区 # 设置时区
TZ=Asia/Shanghai \ TZ=Asia/Shanghai \
# 设置pip镜像地址加快安装速度 # 设置pip镜像地址加快安装速度
PIP_INDEX_URL=http://mirrors.aliyun.com/pypi/simple PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple
# 将当前目录内容复制到容器的/app内 # 将当前目录内容复制到容器的/app内
COPY . /app COPY . /app
# 安装任何需要的包 # 安装任何需要的包
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \ 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 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/ && pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/

View File

@@ -1,3 +1,5 @@
pyyaml==6.0.1
numpy==1.26.4 numpy==1.26.4
onnxconverter-common==1.14.0 onnxconverter-common==1.14.0
opencv-python==4.6.0.66 opencv-python==4.6.0.66