Skip to content

GitLab

Unverified Commit 593a5634 authored by OlivierDehaene's avatar OlivierDehaene Committed by GitHub
Browse files

feat(docker): add nvidia env vars (#255)

parent f092ba9b
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
Dockerfile
View file @ 593a5634
...@@ -164,6 +164,11 @@ FROM base as sagemaker ...@@ -164,6 +164,11 @@ FROM base as sagemaker
COPY sagemaker-entrypoint.sh entrypoint.sh COPY sagemaker-entrypoint.sh entrypoint.sh
RUN chmod +x entrypoint.sh RUN chmod +x entrypoint.sh
# NVIDIA env vars
ENV NVIDIA_VISIBLE_DEVICES all
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
ENTRYPOINT ["./entrypoint.sh"] ENTRYPOINT ["./entrypoint.sh"]
# Final image # Final image
......
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