diff --git a/Dockerfile b/Dockerfile index ed7961a..f612843 100644 --- a/Dockerfile +++ b/Dockerfile @@ -19,8 +19,8 @@ RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone && 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+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 + && 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 # 将当前目录内容复制到容器的/app内 COPY . /app diff --git a/packages/torchaudio-2.4.0+cu118-cp310-cp310-linux_x86_64.whl b/packages/torchaudio-2.4.0+cu118-cp310-cp310-linux_x86_64.whl new file mode 100644 index 0000000..b703c3d Binary files /dev/null and b/packages/torchaudio-2.4.0+cu118-cp310-cp310-linux_x86_64.whl differ diff --git a/packages/torchvision-0.19.0+cu118-cp310-cp310-linux_x86_64.whl b/packages/torchvision-0.19.0+cu118-cp310-cp310-linux_x86_64.whl new file mode 100644 index 0000000..7b8b7e2 Binary files /dev/null and b/packages/torchvision-0.19.0+cu118-cp310-cp310-linux_x86_64.whl differ