Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

安装pytorch，torch包太大，暂时手动传输

Browse Source
This commit is contained in:
liuyebo
2024-08-22 10:35:28 +08:00
parent 9764755936
commit f6ee62313e
3 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Dockerfile
View File

@@ -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

BIN
packages/torchaudio-2.4.0+cu121-cp310-cp310-linux_x86_64.whl Normal file
View File

Binary file not shown.

BIN
packages/torchvision-0.19.0+cu121-cp310-cp310-linux_x86_64.whl Normal file
View File

Binary file not shown.