mirror of
https://github.com/k4yt3x/video2x.git
synced 2024-12-27 14:39:09 +00:00
added cuda drivers to cuda Dockerfile; updated slim Dockerfile for current version
This commit is contained in:
parent
237f6336ec
commit
0fca5c8e9a
@ -1,7 +1,7 @@
|
||||
# Name: Video2X Dockerfile (CUDA)
|
||||
# Creator: K4YT3X
|
||||
# Date Created: February 3, 2022
|
||||
# Last Modified: February 26, 2022
|
||||
# Last Modified: March 18, 2022
|
||||
|
||||
# stage 1: build the python components into wheels
|
||||
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 \
|
||||
python3-pip python3-dev \
|
||||
python3-opencv python3-pil \
|
||||
mesa-vulkan-drivers ffmpeg \
|
||||
mesa-vulkan-drivers cuda-drivers ffmpeg \
|
||||
&& pip install --no-cache-dir --no-index -f /wheels . \
|
||||
&& apt-get clean \
|
||||
&& rm -rf /wheels /video2x /var/lib/apt/lists/*
|
||||
|
@ -1,20 +1,18 @@
|
||||
# Name: Video2X Dockerfile (Slim Alpine)
|
||||
# Creator: K4YT3X
|
||||
# Date Created: February 1, 2022
|
||||
# Last Modified: February 4, 2022
|
||||
# Last Modified: March 18, 2022
|
||||
|
||||
# stage: build python components into heels
|
||||
FROM docker.io/library/python:3.10.2-alpine3.15 AS builder
|
||||
COPY . /video2x
|
||||
WORKDIR /video2x
|
||||
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 \
|
||||
linux-headers \
|
||||
&& pip install -U pip \
|
||||
&& CMAKE_ARGS="-DWITH_FFMPEG=YES" pip wheel -w /wheels opencv-python \
|
||||
&& pip wheel -w /wheels wheel setuptools setuptools_scm \
|
||||
rife-ncnn-vulkan-python@git+https://github.com/media2x/rife-ncnn-vulkan-python.git .
|
||||
&& pip wheel -w /wheels wheel pdm-pep517 .
|
||||
|
||||
# stage 2: install wheels into final image
|
||||
FROM docker.io/library/python:3.10.2-alpine3.15
|
||||
|
Loading…
Reference in New Issue
Block a user