art-dag/celery
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add NVDEC headers and libnvcuvid stub for decord build
Some checks are pending
GPU Worker CI/CD / test (push) Waiting to run
Details
GPU Worker CI/CD / deploy (push) Blocked by required conditions
Details

Browse Source
This commit is contained in:
giles
2026-02-04 01:53:37 +00:00
parent 3a02fca7fd
commit b6292268fa
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
Dockerfile.gpu
View File

@@ -24,6 +24,13 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
&& ln -sf /usr/bin/python3.11 /usr/bin/python3 \ && ln -sf /usr/bin/python3.11 /usr/bin/python3 \
&& ln -sf /usr/bin/python3 /usr/bin/python && ln -sf /usr/bin/python3 /usr/bin/python
# Download Video Codec SDK headers for NVDEC/NVCUVID
RUN git clone https://github.com/FFmpeg/nv-codec-headers.git /tmp/nv-codec-headers && \
cd /tmp/nv-codec-headers && make install && rm -rf /tmp/nv-codec-headers
# Create stub for libnvcuvid (real library comes from driver at runtime)
RUN echo 'void* __nvcuvid_stub__;' | gcc -shared -x c - -o /usr/local/cuda/lib64/libnvcuvid.so
# Build decord with CUDA support # Build decord with CUDA support
RUN git clone --recursive https://github.com/dmlc/decord /tmp/decord && \ RUN git clone --recursive https://github.com/dmlc/decord /tmp/decord && \
cd /tmp/decord && \ cd /tmp/decord && \