diff --git a/Dockerfile.cuda b/Dockerfile.cuda new file mode 100644 index 0000000..3d3b26a --- /dev/null +++ b/Dockerfile.cuda @@ -0,0 +1,13 @@ +FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 + +WORKDIR /subgen + +RUN apt-get update && apt-get -y install python3 python3-pip ffmpeg + +ENV PYTHONUNBUFFERED 1 + +ADD https://raw.githubusercontent.com/McCloudS/subgen/main/subgen/subgen.py /subgen/subgen.py + +CMD [ "python3", "-u", "./subgen.py" ] + +EXPOSE 8090