Merge pull request #441 from containers/fix-vulkan-container-image-build

Only do dnf install for cuda images
containers · Nov 11, 2024 · 76afd2d · 76afd2d
2 parents ba9be31 + 89236dc
commit 76afd2d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/container-images/scripts/build_llama_and_whisper.sh b/container-images/scripts/build_llama_and_whisper.sh
@@ -31,9 +31,12 @@ dnf_install() {
       "${rpm_list[@]}"
   elif [ "$containerfile" = "rocm" ]; then
     dnf install -y rocm-dev hipblas-devel rocblas-devel
-  else
+  elif [ "$containerfile" = "cuda" ]; then
     dnf install -y "${rpm_list[@]}"
   fi
+
+  # For Vulkan image, we don't need to install anything extra but rebuild with
+  # -DGGML_VULKAN
 }
 
 cmake_steps() {