[
https://issues.apache.org/jira/browse/MESOS-8758?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=16425228#comment-16425228
]
Kevin Klues commented on MESOS-8758:
------------------------------------
This is the nvidia-docker documentation about this.
https://github.com/NVIDIA/nvidia-docker/wiki/Image-inspection-%28version-1.0%29
In the past when questions have come up I've sent the following to the mailing
list:
{noformat}
Just deriving from "ubuntu:16.04" wont work out of the box.
In order for the libs to be mounted, and available in the path, the docker
image needs to have:
1) A label set for "com.nvidia.volumes.needed = nvidia_driver"
2) The env set for
"LD_LIBRARY_PATH=/usr/local/nvidia/lib;/usr/local/nvidia/lib64"
3) The env set for "PATH=/usr/local/nvidia/bin;$PATH"
The nvidia images do this automatically as described here:
https://github.com/NVIDIA/nvidia-docker/wiki/Internals
{noformat}
In general, we've always just referred people to the nvidia-docker
documentation since that is the source of truth for what is required. We should
probably duplicate this in our docs though. Especially since (for more than a
year now) nvidia-docker has a new ay of doing things and we've never had the
opportunity to catch up with them. That is, they don't actually need this label
or env vars set anymore. Luckily they stay backwards compatible with their old
way of doing things though, so these labels and envvars are still set in all of
their images. It's very risky to keep assuming they will do this forever
though. My guess is that they will stop doing it within the next year.
> Document LABEL and ENV VAR dependency for GPU support.
> ------------------------------------------------------
>
> Key: MESOS-8758
> URL: https://issues.apache.org/jira/browse/MESOS-8758
> Project: Mesos
> Issue Type: Documentation
> Components: documentation
> Reporter: Gilbert Song
> Priority: Major
> Labels: document
>
> If using GPU on UCR with non-Nvidia based images, the following LABEL and ENV
> VAR is needed in the image:
> {noformat}
> LABEL com.nvidia.volumes.needed="nvidia_driver"
> LABEL com.nvidia.cuda.version="${CUDA_VERSION}"
> {noformat}
> {noformat}
> ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
> ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
> {noformat}
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)
