Skip to content

Commit 2b2934e

Browse files
committed
Extending LD_LIBRARY_PATH before OpenCV build
1 parent 5f8d918 commit 2b2934e

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

ubuntu20.04/Dockerfile

+6-4
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,9 @@ RUN mkdir -p /usr/local/src/builder \
237237
&& rm -rf /usr/local/src/builder
238238
# From https://docs.nvidia.com/video-technologies/video-codec-sdk/ffmpeg-with-nvidia-gpu/#basic-testing
239239
# GPU Testing: ffmpeg -y -vsync 0 -hwaccel cuda -hwaccel_output_format cuda -i input.mp4 -c:a copy -c:v h264_nvenc -b:v 5M output.mp4
240+
# GPU Testing: ffmpeg -y -vsync 0 -hwaccel cuvid -c:v h264_cuvid -i in.mp4 -c:v hevc_nvenc out.mkv
240241

242+
ENV LD_LIBRARY_PATH /usr/local/lib:$LD_LIBRARY_PATH
241243

242244
##### OpenCV
243245

@@ -352,10 +354,7 @@ RUN mkdir -p /usr/local/src \
352354
&& rm -rf /root/.cache/pip /usr/local/src/pytorch
353355
RUN python3 -c "import torch"
354356

355-
ENV LD_LIBRARY_PATH /usr/local/lib:$LD_LIBRARY_PATH
356-
ENV NVIDIA_VISIBLE_DEVICES all
357-
ENV NVIDIA_DRIVER_CAPABILITIES video,compute,utility
358-
357+
# Note: NOT building with Video Codec SDK as it requires an Nvidia account
359358
ARG CTO_TORCHVISION=0.12
360359
RUN mkdir -p /usr/local/src \
361360
&& cd /usr/local/src \
@@ -413,6 +412,9 @@ RUN chmod +x /tmp/tf_info.sh \
413412
RUN mkdir /dmc
414413
WORKDIR /dmc
415414

415+
#ENV NVIDIA_VISIBLE_DEVICES all
416+
#ENV NVIDIA_DRIVER_CAPABILITIES video,compute,utility
417+
416418
CMD bash
417419

418420
LABEL "Author"="Data Machines Corp <[email protected]>"

0 commit comments

Comments
 (0)