File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed
Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff 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
353355RUN 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
359358ARG CTO_TORCHVISION=0.12
360359RUN mkdir -p /usr/local/src \
361360 && cd /usr/local/src \
@@ -413,6 +412,9 @@ RUN chmod +x /tmp/tf_info.sh \
413412RUN mkdir /dmc
414413WORKDIR /dmc
415414
415+ # ENV NVIDIA_VISIBLE_DEVICES all
416+ # ENV NVIDIA_DRIVER_CAPABILITIES video,compute,utility
417+
416418CMD bash
417419
418420LABEL "Author" =
"Data Machines Corp <[email protected] >"
You can’t perform that action at this time.
0 commit comments