install.sh 7.32 KB
Newer Older
1
2
3
4
5
6
#!/bin/sh
# This script installs Ollama on Linux.
# It detects the current operating system architecture and installs the appropriate version of Ollama.

set -eu

7
8
status() { echo ">>> $*" >&2; }
error() { echo "ERROR $*"; exit 1; }
9
warning() { echo "WARNING: $*"; }
10

11
12
13
TEMP_DIR=$(mktemp -d)
cleanup() { rm -rf $TEMP_DIR; }
trap cleanup EXIT
14

Michael Yang's avatar
Michael Yang committed
15
16
available() { command -v $1 >/dev/null; }
require() {
17
18
    local MISSING=''
    for TOOL in $*; do
Michael Yang's avatar
Michael Yang committed
19
        if ! available $TOOL; then
20
            MISSING="$MISSING $TOOL"
21
        fi
22
23
24
    done

    echo $MISSING
25
26
}

27
[ "$(uname -s)" = "Linux" ] || error 'This script is intended to run on Linux only.'
28

29
30
31
32
33
case "$(uname -m)" in
    x86_64) ARCH="amd64" ;;
    aarch64|arm64) ARCH="arm64" ;;
    *) error "Unsupported architecture: $ARCH" ;;
esac
34

35
SUDO=
36
37
if [ "$(id -u)" -ne 0 ]; then
    # Running as root, no need for sudo
Michael Yang's avatar
Michael Yang committed
38
    if ! available sudo; then
Michael Yang's avatar
Michael Yang committed
39
        error "This script requires superuser permissions. Please re-run as root."
40
    fi
41
42

    SUDO="sudo"
43
fi
44

Michael Yang's avatar
Michael Yang committed
45
46
NEEDS=$(require curl awk grep sed tee xargs)
if [ -n "$NEEDS" ]; then
Michael Yang's avatar
Michael Yang committed
47
    status "ERROR: The following tools are required but missing:"
Michael Yang's avatar
Michael Yang committed
48
49
    for NEED in $NEEDS; do
        echo "  - $NEED"
Michael Yang's avatar
Michael Yang committed
50
51
    done
    exit 1
52
53
54
fi

status "Downloading ollama..."
Michael Yang's avatar
Michael Yang committed
55
curl --fail --show-error --location --progress-bar -o $TEMP_DIR/ollama "https://ollama.ai/download/ollama-linux-$ARCH"
56

57
58
59
60
61
62
63
for BINDIR in /usr/local/bin /usr/bin /bin; do
    echo $PATH | grep -q $BINDIR && break || continue
done

status "Installing ollama to $BINDIR..."
$SUDO install -o0 -g0 -m755 -d $BINDIR
$SUDO install -o0 -g0 -m755 $TEMP_DIR/ollama $BINDIR/ollama
64
65
66
67
68

install_success() { status 'Install complete. Run "ollama" from the command line.'; }
trap install_success EXIT

# Everything from this point onwards is optional.
69
70

configure_systemd() {
71
72
    if ! id ollama >/dev/null 2>&1; then
        status "Creating ollama user..."
73
        $SUDO useradd -r -s /bin/false -m -d /usr/share/ollama ollama
74
    fi
75

76
    status "Creating ollama systemd service..."
77
    cat <<EOF | $SUDO tee /etc/systemd/system/ollama.service >/dev/null
78
79
80
81
82
[Unit]
Description=Ollama Service
After=network-online.target

[Service]
83
ExecStart=$BINDIR/ollama serve
84
85
86
87
User=ollama
Group=ollama
Restart=always
RestartSec=3
88
Environment="HOME=/usr/share/ollama"
Michael Yang's avatar
Michael Yang committed
89
Environment="PATH=$PATH"
90
91
92
93

[Install]
WantedBy=default.target
EOF
94
    SYSTEMCTL_RUNNING="$(systemctl is-system-running || true)"
Michael Yang's avatar
Michael Yang committed
95
96
97
98
99
    case $SYSTEMCTL_RUNNING in
        running|degraded)
            status "Enabling and starting ollama service..."
            $SUDO systemctl daemon-reload
            $SUDO systemctl enable ollama
100
101
102

            start_service() { $SUDO systemctl restart ollama; }
            trap start_service EXIT
Michael Yang's avatar
Michael Yang committed
103
104
            ;;
    esac
105
106
}

Michael Yang's avatar
Michael Yang committed
107
if available systemctl; then
108
    configure_systemd
109
110
fi

Michael Yang's avatar
Michael Yang committed
111
112
113
114
115
if ! available lspci && ! available lshw; then
    warning "Unable to detect NVIDIA GPU. Install lspci or lshw to automatically detect and install NVIDIA CUDA drivers."
    exit 0
fi

116
117
check_gpu() {
    case $1 in
Michael Yang's avatar
Michael Yang committed
118
119
120
        lspci) available lspci && lspci -d '10de:' | grep -q 'NVIDIA' || return 1 ;;
        lshw) available lshw && $SUDO lshw -c display -numeric | grep -q 'vendor: .* \[10DE\]' || return 1 ;;
        nvidia-smi) available nvidia-smi || return 1 ;;
121
    esac
122
123
}

124
125
126
127
128
if check_gpu nvidia-smi; then
    status "NVIDIA GPU installed."
    exit 0
fi

129
130
131
132
133
134
135
136
137
138
139
140
141
if ! check_gpu lspci && ! check_gpu lshw; then
    warning "No NVIDIA GPU detected. Ollama will run in CPU-only mode."
    exit 0
fi

# ref: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#rhel-7-centos-7
# ref: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#rhel-8-rocky-8
# ref: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#rhel-9-rocky-9
# ref: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#fedora
install_cuda_driver_yum() {
    status 'Installing NVIDIA repository...'
    case $PACKAGE_MANAGER in
        yum)
142
143
            $SUDO $PACKAGE_MANAGER -y install yum-utils
            $SUDO $PACKAGE_MANAGER-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/$1$2/$(uname -m)/cuda-$1$2.repo
144
145
            ;;
        dnf)
Michael Yang's avatar
Michael Yang committed
146
            $SUDO $PACKAGE_MANAGER config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/$1$2/$(uname -m)/cuda-$1$2.repo
147
148
149
150
151
152
153
            ;;
    esac

    case $1 in
        rhel)
            status 'Installing EPEL repository...'
            # EPEL is required for third-party dependencies such as dkms and libvdpau
154
            $SUDO $PACKAGE_MANAGER -y install https://dl.fedoraproject.org/pub/epel/epel-release-latest-$2.noarch.rpm || true
155
156
157
158
159
160
            ;;
    esac

    status 'Installing CUDA driver...'

    if [ "$1" = 'centos' ] || [ "$1$2" = 'rhel7' ]; then
161
        $SUDO $PACKAGE_MANAGER -y install nvidia-driver-latest-dkms
162
    fi
163
164

    $SUDO $PACKAGE_MANAGER -y install cuda-drivers
165
166
167
168
169
170
171
172
173
174
175
}

# ref: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#ubuntu
# ref: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#debian
install_cuda_driver_apt() {
    status 'Installing NVIDIA repository...'
    curl -fsSL -o $TEMP_DIR/cuda-keyring.deb https://developer.download.nvidia.com/compute/cuda/repos/$1$2/$(uname -m)/cuda-keyring_1.1-1_all.deb

    case $1 in
        debian)
            status 'Enabling contrib sources...'
176
            $SUDO sed 's/main/contrib/' < /etc/apt/sources.list | sudo tee /etc/apt/sources.list.d/contrib.list > /dev/null
177
178
179
180
            ;;
    esac

    status 'Installing CUDA driver...'
181
182
    $SUDO dpkg -i $TEMP_DIR/cuda-keyring.deb
    $SUDO apt-get update
Michael Yang's avatar
Michael Yang committed
183
184
185

    [ -n "$SUDO" ] && SUDO_E="$SUDO -E" || SUDO_E=
    DEBIAN_FRONTEND=noninteractive $SUDO_E apt-get -y install cuda-drivers -q
186
187
188
189
190
191
192
193
194
195
196
197
198
}

if [ ! -f "/etc/os-release" ]; then
    error "Unknown distribution. Skipping CUDA installation."
fi

. /etc/os-release

OS_NAME=$ID
OS_VERSION=$VERSION_ID

PACKAGE_MANAGER=
for PACKAGE_MANAGER in dnf yum apt-get; do
Michael Yang's avatar
Michael Yang committed
199
    if available $PACKAGE_MANAGER; then
200
201
202
203
204
205
206
207
208
209
210
211
        break
    fi
done

if [ -z "$PACKAGE_MANAGER" ]; then
    error "Unknown package manager. Skipping CUDA installation."
fi

if ! check_gpu nvidia-smi || [ -z "$(nvidia-smi | grep -o "CUDA Version: [0-9]*\.[0-9]*")" ]; then
    case $OS_NAME in
        centos|rhel) install_cuda_driver_yum 'rhel' $OS_VERSION ;;
        rocky) install_cuda_driver_yum 'rhel' $(echo $OS_VERSION | cut -c1) ;;
Michael Yang's avatar
Michael Yang committed
212
        fedora) install_cuda_driver_yum $OS_NAME $OS_VERSION ;;
Michael Yang's avatar
Michael Yang committed
213
        amzn) install_cuda_driver_yum 'fedora' '35' ;;
Michael Yang's avatar
Michael Yang committed
214
215
        debian) install_cuda_driver_apt $OS_NAME $OS_VERSION ;;
        ubuntu) install_cuda_driver_apt $OS_NAME $(echo $OS_VERSION | sed 's/\.//') ;;
Michael Yang's avatar
Michael Yang committed
216
        *) exit ;;
217
218
219
220
221
222
    esac
fi

if ! lsmod | grep -q nvidia; then
    KERNEL_RELEASE="$(uname -r)"
    case $OS_NAME in
Bruce MacDonald's avatar
Bruce MacDonald committed
223
224
        centos|rhel|rocky|amzn) $SUDO $PACKAGE_MANAGER -y install kernel-devel-$KERNEL_RELEASE kernel-headers-$KERNEL_RELEASE ;;
        fedora) $SUDO $PACKAGE_MANAGER -y install kernel-devel-$KERNEL_RELEASE ;;
225
        debian|ubuntu) $SUDO apt-get -y install linux-headers-$KERNEL_RELEASE ;;
Michael Yang's avatar
Michael Yang committed
226
        *) exit ;;
227
228
    esac

229
    NVIDIA_CUDA_VERSION=$($SUDO dkms status | awk -F: '/added/ { print $1 }')
Michael Yang's avatar
Michael Yang committed
230
231
232
233
    if [ -n "$NVIDIA_CUDA_VERSION" ]; then
        $SUDO dkms install $NVIDIA_CUDA_VERSION
    fi

Michael Yang's avatar
Michael Yang committed
234
    if lsmod | grep -q nouveau; then
Michael Yang's avatar
Michael Yang committed
235
236
        status 'Reboot to complete NVIDIA CUDA driver install.'
        exit 0
Michael Yang's avatar
Michael Yang committed
237
238
    fi

239
    $SUDO modprobe nvidia
240
fi
241
242


Michael Yang's avatar
Michael Yang committed
243
status "NVIDIA CUDA drivers installed."