diff --git a/Dockerfile b/Dockerfile index c84dcf0..39a4e3c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -17,17 +17,7 @@ COPY packages /app/packages RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \ && 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/ \ - && pip install packages/torch-2.4.0+cu118-cp310-cp310-linux_x86_64.whl \ - && pip install packages/torchvision-0.19.0+cu118-cp310-cp310-linux_x86_64.whl \ - && pip install packages/torchaudio-2.4.0+cu118-cp310-cp310-linux_x86_64.whl - -# 编译文档检测的部分内容 -COPY object_detection /app/object_detection -RUN cd /app/object_detection/core/models/py_utils/_cpools \ - && python setup.py install --user \ - && cd /app/object_detection/core/external \ - && make + && pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ # 将当前目录内容复制到容器的/app内 COPY . /app