docker-compose.yml 702 B

12345678910111213141516171819
  1. ---
  2. version: '3'
  3. services:
  4. nvidia_smi_exporter: #To export data from nvidia-smi, needs nvidia-smi and nvidia-container-toolkit installed on host.
  5. image: utkuozdemir/nvidia_gpu_exporter:0.3.0
  6. container_name: nvidia_smi_exporter
  7. runtime: nvidia
  8. environment:
  9. - NVIDIA_VISIBLE_DEVICES=all
  10. - NVIDIA_DRIVER_CAPABILITIES=all
  11. ports:
  12. - "9835:9835"
  13. volumes:
  14. - /usr/bin/nvidia-smi:/usr/bin/nvidia-smi
  15. - /usr/lib/x86_64-linux-gnu/libnvidia-ml.so:/usr/lib/x86_64-linux-gnu/libnvidia-ml.so
  16. - /usr/lib/x86_64-linux-gnu/libnvidia-ml.so.1:/usr/lib/x86_64-linux-gnu/libnvidia-ml.so.1
  17. restart: unless-stopped
  18. #Mount points for volume work on Ubuntu 20.04