baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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 esac
} }
if check_gpu nvidia-smi; then
status "NVIDIA GPU installed."
exit 0
fi
if ! check_gpu lspci && ! check_gpu lshw; then if ! check_gpu lspci && ! check_gpu lshw; then
warning "No NVIDIA GPU detected. Ollama will run in CPU-only mode." warning "No NVIDIA GPU detected. Ollama will run in CPU-only mode."
exit 0 exit 0
@ -228,3 +233,6 @@ if ! lsmod | grep -q nvidia; then
$SUDO modprobe nvidia $SUDO modprobe nvidia
fi fi
status "NVIDIA GPU installed."