mirrors/ollama-for-amd
1
0
Fork 0
mirror of https://github.com/likelovewant/ollama-for-amd.git synced 2025-12-22 06:43:57 +00: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
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."