Use Vulkan

scripts/local.sh

@@ -21,6 +21,13 @@ docker build \
     --tag sia \
     .
 
+GPU_FLAGS=""
+# Check if AMD GPU devices are available
+if [ -e /dev/dri ] && [ -e /dev/kfd ]; then
+    echo "AMD GPU detected, enabling GPU acceleration"
+    GPU_FLAGS="--device=/dev/dri --device=/dev/kfd --group-add video"
+fi
+
 docker run \
     --init \
     --rm \
@@ -28,6 +35,7 @@ docker run \
     -p 8080:8080 \
     -p 8000:8000 \
     --env-file .env \
+    $GPU_FLAGS \
     -v /$(pwd)/models/:/root/models/ \
     -v /$(pwd)/iterations/:/root/data/iterations/ \
     -v /$(pwd)/tasks/:/root/data/tasks/ \