dockerfile: add some reasonable ENV defaults

This commit is contained in:
Vladislav Yarmak 2019-07-11 03:26:23 +03:00
parent 94222a8824
commit 389235cd92

View File

@ -1,5 +1,8 @@
FROM nvidia/cuda:latest FROM nvidia/cuda:latest
ENV NVIDIA_VISIBLE_DEVICES all
ENV NVIDIA_DRIVER_CAPABILITIES compute,video,utility
RUN mkdir -p /usr/local/bin /patched-lib RUN mkdir -p /usr/local/bin /patched-lib
COPY patch.sh docker-entrypoint.sh /usr/local/bin/ COPY patch.sh docker-entrypoint.sh /usr/local/bin/
RUN chmod +x /usr/local/bin/patch.sh /usr/local/bin/docker-entrypoint.sh RUN chmod +x /usr/local/bin/patch.sh /usr/local/bin/docker-entrypoint.sh