From 82b742f8d7ac8911bfd27324b4f13914fdf7b18b Mon Sep 17 00:00:00 2001 From: McCloudS <64094529+McCloudS@users.noreply.github.com> Date: Mon, 23 Oct 2023 14:21:59 -0600 Subject: [PATCH] Create Dockerfile.cuda --- Dockerfile.cuda | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 Dockerfile.cuda 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