diff --git a/build-image.sh b/build-image.sh index ce7b057..2a6a1ef 100755 --- a/build-image.sh +++ b/build-image.sh @@ -138,6 +138,12 @@ apt-get install -y \ containerd \ runc +# Vulkan drivers for GPU compute workloads (ollama, llama.cpp) +apt-get install -y \ + mesa-vulkan-drivers \ + libvulkan1 \ + vulkan-tools + # Useful tools apt-get install -y \ htop \