diff --git a/docker-compose.yml b/docker-compose.yml
index 6910581..a3a6aab 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -19,13 +19,18 @@ services:
     runtime: nvidia
 
   ollama-exporter:
-    image: ghcr.io/frcooper/ollama-exporter:latest
+    image: ghcr.io/mikeh-22/ollama-exporter:latest
     container_name: ollama-exporter
     restart: unless-stopped
-    ports:
-      - "8000:8000"
+    runtime: nvidia
     environment:
-      - OLLAMA_ENDPOINT=http://ollama:11434
+      - NVIDIA_VISIBLE_DEVICES=all
+      - OLLAMA_URL=http://ollama:11434
+    ports:
+      - "9101:9101"
+    volumes:
+      - /sys:/sys:ro
+      - /var/run/docker.sock:/var/run/docker.sock
     depends_on:
       - ollama