File tree 1 file changed +6
-4
lines changed
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 \
237
237
&& rm -rf /usr/local/src/builder
238
238
# From https://docs.nvidia.com/video-technologies/video-codec-sdk/ffmpeg-with-nvidia-gpu/#basic-testing
239
239
# 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
240
241
242
+ ENV LD_LIBRARY_PATH /usr/local/lib:$LD_LIBRARY_PATH
241
243
242
244
# #### OpenCV
243
245
@@ -352,10 +354,7 @@ RUN mkdir -p /usr/local/src \
352
354
&& rm -rf /root/.cache/pip /usr/local/src/pytorch
353
355
RUN python3 -c "import torch"
354
356
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
359
358
ARG CTO_TORCHVISION=0.12
360
359
RUN mkdir -p /usr/local/src \
361
360
&& cd /usr/local/src \
@@ -413,6 +412,9 @@ RUN chmod +x /tmp/tf_info.sh \
413
412
RUN mkdir /dmc
414
413
WORKDIR /dmc
415
414
415
+ # ENV NVIDIA_VISIBLE_DEVICES all
416
+ # ENV NVIDIA_DRIVER_CAPABILITIES video,compute,utility
417
+
416
418
CMD bash
417
419
418
420
LABEL "Author" =
"Data Machines Corp <[email protected] >"
You can’t perform that action at this time.
0 commit comments