Ma/video2x
1
0
Fork 0
mirror of https://github.com/k4yt3x/video2x.git synced 2026-02-19 04:55:48 +08:00
Code Issues 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
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)
# 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/*