mirror of
https://github.com/YaoFANGUK/video-subtitle-remover.git
synced 2026-02-04 04:34:41 +08:00
Revert "由于PySimpleGUI作者故意移除免费的旧版本,改用PySimpleGUI-4-foss"
This commit is contained in:
@@ -1,30 +0,0 @@
|
||||
FROM python:3.12
|
||||
|
||||
RUN --mount=type=cache,target=/root/.cache,sharing=private \
|
||||
apt update && \
|
||||
apt install -y libgl1-mesa-glx && \
|
||||
true
|
||||
|
||||
ADD . /vsr
|
||||
ARG CUDA_VERSION=11.8
|
||||
ARG USE_DIRECTML=0
|
||||
|
||||
# 如果是 CUDA 版本,执行 CUDA 特定设置
|
||||
RUN --mount=type=cache,target=/root/.cache,sharing=private \
|
||||
if [ "${USE_DIRECTML:-0}" != "1" ]; then \
|
||||
pip install paddlepaddle==3.0 && \
|
||||
pip install torch==2.7.0 torchvision==0.22.0 --index-url https://download.pytorch.org/whl/cu$(echo ${CUDA_VERSION} | tr -d '.') && \
|
||||
pip install -r /vsr/requirements.txt; \
|
||||
fi
|
||||
|
||||
# 如果是 DirectML 版本,执行 DirectML 特定设置
|
||||
RUN --mount=type=cache,target=/root/.cache,sharing=private \
|
||||
if [ "${USE_DIRECTML:-0}" = "1" ]; then \
|
||||
pip install paddlepaddle==3.0 && \
|
||||
pip install torch_directml==0.2.5.dev240914 && \
|
||||
pip install -r /vsr/requirements.txt; \
|
||||
fi
|
||||
|
||||
ENV LD_LIBRARY_PATH=/usr/local/lib/python3.12/site-packages/nvidia/cudnn/lib/
|
||||
WORKDIR /vsr
|
||||
CMD ["python", "/vsr/backend/main.py"]
|
||||
Reference in New Issue
Block a user