|
@@ -93,6 +93,22 @@ else
|
|
|
fi
|
|
|
fi
|
|
|
|
|
|
+# Check for NVIDIA JetPack systems with additional downloads
|
|
|
+if [ -f /etc/nv_tegra_release ] ; then
|
|
|
+ if grep R36 /etc/nv_tegra_release > /dev/null ; then
|
|
|
+ status "Downloading JetPack 6 components"
|
|
|
+ curl --fail --show-error --location --progress-bar \
|
|
|
+ "https://ollama.com/download/ollama-linux-${ARCH}-jetpack6.tgz${VER_PARAM}" | \
|
|
|
+ $SUDO tar -xzf - -C "$OLLAMA_INSTALL_DIR"
|
|
|
+ elif grep R35 /etc/nv_tegra_release > /dev/null ; then
|
|
|
+ status "Downloading JetPack 5 components"
|
|
|
+ curl --fail --show-error --location --progress-bar \
|
|
|
+ "https://ollama.com/download/ollama-linux-${ARCH}-jetpack5.tgz${VER_PARAM}" | \
|
|
|
+ $SUDO tar -xzf - -C "$OLLAMA_INSTALL_DIR"
|
|
|
+ else
|
|
|
+ warning "Unsupported JetPack version detected. GPU may not be supported"
|
|
|
+ fi
|
|
|
+fi
|
|
|
|
|
|
install_success() {
|
|
|
status 'The Ollama API is now available at 127.0.0.1:11434.'
|
|
@@ -163,6 +179,13 @@ if [ "$IS_WSL2" = true ]; then
|
|
|
exit 0
|
|
|
fi
|
|
|
|
|
|
+# Don't attempt to install drivers on Jetson systems
|
|
|
+if [ -f /etc/nv_tegra_release ] ; then
|
|
|
+ status "NVIDIA JetPack ready."
|
|
|
+ install_success
|
|
|
+ exit 0
|
|
|
+fi
|
|
|
+
|
|
|
# Install GPU dependencies on Linux
|
|
|
if ! available lspci && ! available lshw; then
|
|
|
warning "Unable to detect NVIDIA/AMD GPU. Install lspci or lshw to automatically detect and install GPU dependencies."
|