diff --git a/Dockerfile b/Dockerfile index 0d0fc83..ed7961a 100644 --- a/Dockerfile +++ b/Dockerfile @@ -13,10 +13,14 @@ ENV PYTHONUNBUFFERED=1 \ # 安装依赖 COPY requirements.txt /app/requirements.txt +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 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+cu121-cp310-cp310-linux_x86_64.whl \ + && pip install packages/torchvision-0.19.0+cu121-cp310-cp310-linux_x86_64.whl \ + && pip install packages/torchaudio-2.4.0+cu121-cp310-cp310-linux_x86_64.whl # 将当前目录内容复制到容器的/app内 COPY . /app diff --git a/packages/torchaudio-2.4.0+cu121-cp310-cp310-linux_x86_64.whl b/packages/torchaudio-2.4.0+cu121-cp310-cp310-linux_x86_64.whl new file mode 100644 index 0000000..2b1e2d4 Binary files /dev/null and b/packages/torchaudio-2.4.0+cu121-cp310-cp310-linux_x86_64.whl differ diff --git a/packages/torchvision-0.19.0+cu121-cp310-cp310-linux_x86_64.whl b/packages/torchvision-0.19.0+cu121-cp310-cp310-linux_x86_64.whl new file mode 100644 index 0000000..c44fca1 Binary files /dev/null and b/packages/torchvision-0.19.0+cu121-cp310-cp310-linux_x86_64.whl differ