AMD IGPU support (#14944)

This commit is contained in:
Nick B
2026-06-05 06:40:26 -04:00
committed by GitHub
parent 416717eeb1
commit 79ccc8ed6b
2 changed files with 12 additions and 1 deletions

View File

@@ -69,6 +69,13 @@ if ! fetch_and_deploy_gh_release "ollama-com" "ollama/ollama" "prebuild" "latest
msg_error "Failed to download or deploy Ollama check network connectivity and GitHub API availability"
exit 250
fi
# If /dev/kfd exists assume an AMD GPU is installed, and install ROCM support for ollama
if [[ -e /dev/kfd ]]; then
if ! fetch_and_deploy_gh_release "ollama-rocm-com" "ollama/ollama" "prebuild" "latest" "$OLLAMA_INSTALL_DIR/lib" "ollama-linux-amd64-rocm.tar.zst"; then
msg_error "Failed to download or deploy Ollama AMD ROCM suport check network connectivity and GitHub API availability"
exit 250
fi
fi
ln -sf "$OLLAMA_INSTALL_DIR/bin/ollama" "$BINDIR/ollama"
msg_ok "Installed Ollama"
@@ -102,6 +109,10 @@ RestartSec=3
[Install]
WantedBy=multi-user.target
EOF
if [[ -e /dev/kfd ]]; then
sed -i '/Environment=OLLAMA_INTEL_GPU=true/a Environment=OLLAMA_IGPU_ENABLE=1' \
/etc/systemd/system/ollama.service
fi
systemctl enable -q --now ollama
msg_ok "Created Service"

View File

@@ -3982,7 +3982,7 @@ $PCT_OPTIONS_STRING"
if [[ -d /dev/dri ]]; then
# Only add if not already claimed by Intel
if [[ ${#INTEL_DEVICES[@]} -eq 0 ]]; then
for d in /dev/dri/renderD* /dev/dri/card*; do
for d in /dev/dri/renderD* /dev/dri/card* /dev/kfd; do
[[ -e "$d" ]] && AMD_DEVICES+=("$d")
done
fi