docker latest deps: added nvidia runtime stuff for convenience

This commit is contained in:
matlabbe
2023-03-05 16:38:09 -08:00
parent 8c5c4f63bf
commit d6704ca2ed

View File

@@ -61,3 +61,9 @@ RUN source /ros_entrypoint.sh && \
rm -rf rtabmap && \
ldconfig
# nvidia-container-runtime
ENV NVIDIA_VISIBLE_DEVICES \
${NVIDIA_VISIBLE_DEVICES:-all}
ENV NVIDIA_DRIVER_CAPABILITIES \
${NVIDIA_DRIVER_CAPABILITIES:+$NVIDIA_DRIVER_CAPABILITIES,}graphics