Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
ollama
Commits
5e388f93
Commit
5e388f93
authored
Sep 25, 2023
by
Michael Yang
Browse files
check cuda installed before installing
parent
d5ad41dd
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
0 deletions
+8
-0
scripts/install.sh
scripts/install.sh
+8
-0
No files found.
scripts/install.sh
View file @
5e388f93
...
...
@@ -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."
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment