OpenSourceJoker/video2x
3
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2024-12-28 23:19:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added cuda drivers to cuda Dockerfile; updated slim Dockerfile for current version

Browse Source
This commit is contained in:
k4yt3x 2022-03-18 23:10:40 +00:00
parent 237f6336ec
commit 0fca5c8e9a
2 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile.cuda
View File

@ -1,7 +1,7 @@
# Name: Video2X Dockerfile (CUDA) # Name: Video2X Dockerfile (CUDA)
# Creator: K4YT3X # Creator: K4YT3X
# Date Created: February 3, 2022 # Date Created: February 3, 2022
# Last Modified: February 26, 2022 # Last Modified: March 18, 2022
# stage 1: build the python components into wheels # stage 1: build the python components into wheels
FROM docker.io/nvidia/cuda:11.6.0-runtime-ubuntu20.04 AS builder FROM docker.io/nvidia/cuda:11.6.0-runtime-ubuntu20.04 AS builder
@ -30,7 +30,7 @@ WORKDIR /video2x
RUN apt-get install -y --no-install-recommends \ RUN apt-get install -y --no-install-recommends \
python3-pip python3-dev \ python3-pip python3-dev \
python3-opencv python3-pil \ python3-opencv python3-pil \
mesa-vulkan-drivers ffmpeg \ mesa-vulkan-drivers cuda-drivers ffmpeg \
&& pip install --no-cache-dir --no-index -f /wheels . \ && pip install --no-cache-dir --no-index -f /wheels . \
&& apt-get clean \ && apt-get clean \
&& rm -rf /wheels /video2x /var/lib/apt/lists/* && rm -rf /wheels /video2x /var/lib/apt/lists/*

8
Dockerfile.slim-alpine
View File

@ -1,20 +1,18 @@
# Name: Video2X Dockerfile (Slim Alpine) # Name: Video2X Dockerfile (Slim Alpine)
# Creator: K4YT3X # Creator: K4YT3X
# Date Created: February 1, 2022 # Date Created: February 1, 2022
# Last Modified: February 4, 2022 # Last Modified: March 18, 2022
# stage: build python components into heels # stage: build python components into heels
FROM docker.io/library/python:3.10.2-alpine3.15 AS builder FROM docker.io/library/python:3.10.2-alpine3.15 AS builder
COPY . /video2x COPY . /video2x
WORKDIR /video2x WORKDIR /video2x
RUN apk add --no-cache \ RUN apk add --no-cache \
cmake g++ gcc git ninja swig \ cmake g++ gcc git ninja swig linux-headers \
ffmpeg-dev glslang-dev jpeg-dev vulkan-loader-dev zlib-dev \ ffmpeg-dev glslang-dev jpeg-dev vulkan-loader-dev zlib-dev \
linux-headers \
&& pip install -U pip \ && pip install -U pip \
&& CMAKE_ARGS="-DWITH_FFMPEG=YES" pip wheel -w /wheels opencv-python \ && CMAKE_ARGS="-DWITH_FFMPEG=YES" pip wheel -w /wheels opencv-python \
&& pip wheel -w /wheels wheel setuptools setuptools_scm \ && pip wheel -w /wheels wheel pdm-pep517 .
rife-ncnn-vulkan-python@git+https://github.com/media2x/rife-ncnn-vulkan-python.git .
# stage 2: install wheels into final image # stage 2: install wheels into final image
FROM docker.io/library/python:3.10.2-alpine3.15 FROM docker.io/library/python:3.10.2-alpine3.15