This commit is contained in:
parent
381c421fe9
commit
775caac9c0
2 changed files with 5 additions and 4 deletions
|
|
@ -25,6 +25,11 @@ services:
|
||||||
- /usr/lib/x86_64-linux-gnu/libcuda.so.1:/usr/lib/x86_64-linux-gnu/libcuda.so.1:ro
|
- /usr/lib/x86_64-linux-gnu/libcuda.so.1:/usr/lib/x86_64-linux-gnu/libcuda.so.1:ro
|
||||||
- /usr/lib/x86_64-linux-gnu/libcuda.so:/usr/lib/x86_64-linux-gnu/libcuda.so:ro
|
- /usr/lib/x86_64-linux-gnu/libcuda.so:/usr/lib/x86_64-linux-gnu/libcuda.so:ro
|
||||||
- /usr/lib/x86_64-linux-gnu/libnvidia-ml.so.1:/usr/lib/x86_64-linux-gnu/libnvidia-ml.so.1:ro
|
- /usr/lib/x86_64-linux-gnu/libnvidia-ml.so.1:/usr/lib/x86_64-linux-gnu/libnvidia-ml.so.1:ro
|
||||||
|
devices:
|
||||||
|
- /dev/nvidia0:/dev/nvidia0
|
||||||
|
- /dev/nvidiactl:/dev/nvidiactl
|
||||||
|
- /dev/nvidia-uvm:/dev/nvidia-uvm
|
||||||
|
- /dev/nvidia-uvm-tools:/dev/nvidia-uvm-tools
|
||||||
|
|
||||||
deploy:
|
deploy:
|
||||||
resources:
|
resources:
|
||||||
|
|
|
||||||
|
|
@ -21,7 +21,3 @@ services:
|
||||||
- NVIDIA_VISIBLE_DEVICES=all
|
- NVIDIA_VISIBLE_DEVICES=all
|
||||||
- NVIDIA_DRIVER_CAPABILITIES=all,video,compute,utility
|
- NVIDIA_DRIVER_CAPABILITIES=all,video,compute,utility
|
||||||
- LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu
|
- LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu
|
||||||
devices:
|
|
||||||
- /dev/nvidia0:/dev/nvidia0
|
|
||||||
- /dev/nvidiactl:/dev/nvidiactl
|
|
||||||
- /dev/nvidia-uvm:/dev/nvidia-uvm
|
|
||||||
Loading…
Reference in a new issue