Skip to content

N
nvidia-patch
Commit 389235cd authored by Vladislav Yarmak's avatar Vladislav Yarmak
Browse files
Options

dockerfile: add some reasonable ENV defaults

parent 94222a88
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
Dockerfile
View file @ 389235cd
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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment