Skip to content

Commit

Permalink
fix: Dockerfile cuda
Browse files Browse the repository at this point in the history
  • Loading branch information
gedoensmax committed Feb 7, 2024
1 parent 346e176 commit d4eca01
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 4 deletions.
4 changes: 4 additions & 0 deletions .dockerignore
Original file line number Diff line number Diff line change
Expand Up @@ -2,3 +2,7 @@ matlab
workspace
python/.tox
.git
.venv
*.mp4
*.yuv
*.mov
10 changes: 6 additions & 4 deletions Dockerfile.cuda
Original file line number Diff line number Diff line change
@@ -1,18 +1,20 @@
ARG CUDA_VERSION=12.1.0
ARG CUDA_VERSION=12.3.1
ARG VMAF_TAG=master
ARG FFFMPEG_TAG=master
# By copying the installation from a devel to a runtime container one could likely save a lot container size
FROM nvidia/cuda:$CUDA_VERSION-devel-ubuntu22.04

RUN DEBIAN_FRONTEND=noninteractive apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y libopenjp2-7-dev \
ninja-build cmake git python3 python3-pip nasm xxd pkg-config curl unzip

RUN git clone https://github.com/Netflix/vmaf.git
RUN git clone https://github.com/Netflix/vmaf.git && cd vmaf && git checkout $VMAF_TAG

RUN git clone https://github.com/FFmpeg/FFmpeg.git
RUN git clone https://github.com/FFmpeg/FFmpeg.git && cd FFmpeg && git checkout $FFFMPEG_TAG

RUN git clone https://github.com/FFmpeg/nv-codec-headers.git && cd nv-codec-headers && make && make install

# install vmaf
RUN python3 -m pip install meson cpython
RUN python3 -m pip install meson
RUN cd vmaf && meson libvmaf/build libvmaf -Denable_cuda=true -Denable_avx512=true --buildtype release && \
ninja -vC libvmaf/build && \
ninja -vC libvmaf/build install
Expand Down

0 comments on commit d4eca01

Please sign in to comment.