From 9a746597397838df3360f678c21068d75473ca38 Mon Sep 17 00:00:00 2001 From: McCloudS <64094529+McCloudS@users.noreply.github.com> Date: Thu, 26 Oct 2023 01:17:46 -0600 Subject: [PATCH] Update Dockerfile.cuda --- Dockerfile.cuda | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/Dockerfile.cuda b/Dockerfile.cuda index 2198c81..3d3b26a 100644 --- a/Dockerfile.cuda +++ b/Dockerfile.cuda @@ -2,13 +2,12 @@ FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 WORKDIR /subgen -RUN apt-get update && apt-get -y install python3 python3-pip ffmpeg curl +RUN apt-get update && apt-get -y install python3 python3-pip ffmpeg ENV PYTHONUNBUFFERED 1 -ADD https://raw.githubusercontent.com/McCloudS/subgen/main/bootstrap.sh /bootstrap.sh -RUN chmod +x /bootstrap.sh +ADD https://raw.githubusercontent.com/McCloudS/subgen/main/subgen/subgen.py /subgen/subgen.py -CMD [ "/bootstrap.sh" ] +CMD [ "python3", "-u", "./subgen.py" ] EXPOSE 8090