Skip to content

Commit

Permalink
Docker support (#166)
Browse files Browse the repository at this point in the history
* dockerfile up

* rename dockerfile to reflectpurpose
  • Loading branch information
levan92 authored Jul 18, 2021
1 parent 35c87a6 commit 96b750c
Showing 1 changed file with 48 additions and 0 deletions.
48 changes: 48 additions & 0 deletions gpu.Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
FROM nvcr.io/nvidia/cuda:11.2.0-cudnn8-devel-ubuntu20.04

ENV DEBIAN_FRONTEND=noninteractive

RUN apt-get -y update && apt-get install -y \
software-properties-common \
build-essential \
checkinstall \
cmake \
make \
pkg-config \
yasm \
git \
vim \
curl \
wget \
sudo \
apt-transport-https \
libcanberra-gtk-module \
libcanberra-gtk3-module \
dbus-x11 \
iputils-ping \
python3-dev \
python3-pip \
python3-setuptools

# some image/media dependencies
RUN apt-get -y update && apt-get install -y \
libjpeg8-dev \
libpng-dev \
libtiff5-dev \
libtiff-dev \
libavcodec-dev \
libavformat-dev \
libswscale-dev \
libdc1394-22-dev \
libxine2-dev \
libavfilter-dev \
libavutil-dev

RUN apt-get -y update && apt-get install -y ffmpeg

RUN apt-get clean && rm -rf /tmp/* /var/tmp/* /var/lib/apt/lists/* && apt-get -y autoremove

ENV NVIDIA_DRIVER_CAPABILITIES=all
RUN ln -s /usr/lib/x86_64-linux-gnu/libnvcuvid.so.1 /usr/local/cuda/lib64/libnvcuvid.so
RUN git clone --recursive https://github.com/dmlc/decord
RUN cd decord && mkdir build && cd build && cmake .. -DUSE_CUDA=ON -DCMAKE_BUILD_TYPE=Release && make -j2 && cd ../python && python3 setup.py install

2 comments on commit 96b750c

@muhammadumerjaved44
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hey @levan92 libavdevice is missing

also
image

@YuanmingLeee
Copy link

@YuanmingLeee YuanmingLeee commented on 96b750c Jan 26, 2022

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hey @levan92 libavdevice is missing

Try install the libavdevice-dev to solve the missing problem.
At line 39,

...
# line 39 change to:
    libavutil-dev \
    libavdevice-dev
...

also image

Since the NVIDIA runtime is disabled during build time, the libnvcuvid is not linked. Check #102 (comment) to enable Nvidia runtime by default.

Please sign in to comment.