highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-03-26 09:42:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

check cuda installed before installing

Browse Source
This commit is contained in:
Michael Yang 2023-09-25 17:56:43 -07:00
parent d5ad41dd7b
commit 5e388f931e
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/install.sh
View File

@ -117,6 +117,11 @@ check_gpu() {
esac
}
if check_gpu nvidia-smi; then
status "NVIDIA GPU installed."
exit 0
fi
if ! check_gpu lspci && ! check_gpu lshw; then
warning "No NVIDIA GPU detected. Ollama will run in CPU-only mode."
exit 0
@ -228,3 +233,6 @@ if ! lsmod | grep -q nvidia; then
$SUDO modprobe nvidia
fi
status "NVIDIA GPU installed."