dockerfile: add some reasonable ENV defaults

389235cd · Vladislav Yarmak · 94222a88 · 389235cd
Commit 389235cd authored Jul 11, 2019 by Vladislav Yarmak
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

Dockerfile Dockerfile +3 -0

No files found.
--- a/Dockerfile
+++ b/Dockerfile
 FROM nvidia/cuda:latest

+ENV NVIDIA_VISIBLE_DEVICES all
+ENV NVIDIA_DRIVER_CAPABILITIES compute,video,utility
+
 RUN mkdir -p /usr/local/bin /patched-lib
 COPY patch.sh docker-entrypoint.sh /usr/local/bin/
 RUN chmod +x /usr/local/bin/patch.sh /usr/local/bin/docker-entrypoint.sh