NVIDIA / libnvidia-container

NVIDIA container runtime library
Apache License 2.0
843 stars 205 forks source link

Add imex support #242

Closed klueska closed 8 months ago

klueska commented 8 months ago
docker run -it \
        -e NVIDIA_IMEX_CHANNELS=0,1,2047 \
        -e NVIDIA_VISIBLE_DEVICES=0 \
        ubuntu:22.04 bash -c "ls -la /dev/nvidia*"
crw-rw-rw- 1 root root 504,   0 Jan 18 12:59 /dev/nvidia-uvm
crw-rw-rw- 1 root root 504,   1 Jan 18 12:59 /dev/nvidia-uvm-tools
crw-rw-rw- 1 root root 195,   0 Jan 18 12:59 /dev/nvidia0
crw-rw-rw- 1 root root 195, 255 Jan 18 12:59 /dev/nvidiactl

/dev/nvidia-caps-imex-channels:
total 0
drwxr-xr-x 2 root root       100 Jan 18 21:32 .
drwxr-xr-x 6 root root       460 Jan 18 21:32 ..
crw-rw-rw- 1 root root 508,    0 Jan 18 19:11 channel0
crw-rw-rw- 1 root root 508,    1 Jan 18 19:11 channel1
crw-rw-rw- 1 root root 508, 2047 Jan 18 19:13 channel2047