UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2024-11-22 00:08:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
19 Commits 20 Branches 73 Tags 28 MiB
Go 87%
Shell 7.1%
C 3.8%
Makefile 1.7%
Dockerfile 0.3%
Other 0.1%
01b4381282
Go to file
Kevin Klues 01b4381282 Add ability to merge envars of the form NVIDIA_VISIBLE_DEVICES_* 
This allows someone to (for example) pass the following environment
variables:

NVIDIA_VISIBLE_DEVICES_0="0,1"
NVIDIA_VISIBLE_DEVICES_1="2,3"
NVIDIA_VISIBLE_DEVICES_WHATEVER="4,5"

and have the nvidia-container-toolkit automatically merge these into:

NVIDIA_VISIBLE_DEVICES="0,1,2,3,4,5"

This is useful (for example) if the full list of devices comes
from multiple, disparate sources.

Note: This will override whatever the original value of
NVIDIA_VISIBLE_DEVICES was (*excluding* its original value) if it also
exists as an environment variable already. We exclude the original value
to ensure that we have a way to override the default value of
NVIDIA_VISIBLE_DEVICES set to "all" inside a container image.

Signed-off-by: Kevin Klues <kklues@nvidia.com>
2020-05-15 19:04:05 +00:00
config Add ability to merge envars of the form NVIDIA_VISIBLE_DEVICES_* 2020-05-15 19:04:05 +00:00
docker Update build system to match libnvidia-container 2020-05-14 19:27:57 +00:00
packaging Add binary target and use go mod 2020-04-11 17:18:14 -07:00
pkg Add ability to merge envars of the form NVIDIA_VISIBLE_DEVICES_* 2020-05-15 19:04:05 +00:00
vendor Add binary target and use go mod 2020-04-11 17:18:14 -07:00
.dockerignore Add binary target and use go mod 2020-04-11 17:18:14 -07:00
.gitignore Init 2019-10-22 14:36:22 -07:00
.gitlab-ci.yml Update .gitlab-ci.yml to match that of libnvidia-container 2020-05-14 21:42:14 +02:00
container_config_test.go Add ability to merge envars of the form NVIDIA_VISIBLE_DEVICES_* 2020-05-15 19:04:05 +00:00
CONTRIBUTING.md Ensure LICENSE and CONTRIBUTING.md files are present 2019-10-31 12:56:46 -07:00
docker.mk Update build system to match libnvidia-container 2020-05-14 19:27:57 +00:00
go.mod Add binary target and use go mod 2020-04-11 17:18:14 -07:00
go.sum Add binary target and use go mod 2020-04-11 17:18:14 -07:00
LICENSE Ensure LICENSE and CONTRIBUTING.md files are present 2019-10-31 12:56:46 -07:00
Makefile Update build system to match libnvidia-container 2020-05-14 19:27:57 +00:00
oci-nvidia-hook Init 2019-10-22 14:36:22 -07:00
oci-nvidia-hook.json Init 2019-10-22 14:36:22 -07:00