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:/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
|
||||
devices:
|
||||
- /dev/nvidia0:/dev/nvidia0
|
||||
- /dev/nvidiactl:/dev/nvidiactl
|
||||
- /dev/nvidia-uvm:/dev/nvidia-uvm
|
||||
- /dev/nvidia-uvm-tools:/dev/nvidia-uvm-tools
|
||||
|
||||
deploy:
|
||||
resources:
|
||||
|
|
|
|||
|
|
@ -21,7 +21,3 @@ services:
|
|||
- NVIDIA_VISIBLE_DEVICES=all
|
||||
- NVIDIA_DRIVER_CAPABILITIES=all,video,compute,utility
|
||||
- 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