Closed zeryx closed 4 years ago
This PR adds nvidia/cuda support to docker runtimes when requested by the user. Simply add the --nvidia-support flag, and ensure that you're base image extends/is nvidia/cuda:xxx.
--nvidia-support
nvidia/cuda:xxx
rebased from master as it got out of sync.
This PR adds nvidia/cuda support to docker runtimes when requested by the user. Simply add the
--nvidia-support
flag, and ensure that you're base image extends/isnvidia/cuda:xxx
.