docker.mdx 2.22 KB
Newer Older
1
## CPU only
Michael Yang's avatar
Michael Yang committed
2

3
```shell
Michael Yang's avatar
Michael Yang committed
4
5
6
docker run -d -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
```

7
8
## Nvidia GPU

Michael Yang's avatar
Michael Yang committed
9
10
Install the [NVIDIA Container Toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html#installation).

11
12
### Install with Apt

Michael Yang's avatar
Michael Yang committed
13
1.  Configure the repository
14
15
16
17

    ```shell
    curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey \
        | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
18
    curl -fsSL https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list \
19
20
21
22
23
        | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' \
        | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
    sudo apt-get update
    ```

Michael Yang's avatar
Michael Yang committed
24
2.  Install the NVIDIA Container Toolkit packages
25
26
27
28

    ```shell
    sudo apt-get install -y nvidia-container-toolkit
    ```
Michael Yang's avatar
Michael Yang committed
29

30
31
### Install with Yum or Dnf

Michael Yang's avatar
Michael Yang committed
32
33
1.  Configure the repository

34
    ```shell
35
    curl -fsSL https://nvidia.github.io/libnvidia-container/stable/rpm/nvidia-container-toolkit.repo \
36
37
        | sudo tee /etc/yum.repos.d/nvidia-container-toolkit.repo
    ```
Michael Yang's avatar
Michael Yang committed
38

39
2.  Install the NVIDIA Container Toolkit packages
Michael Yang's avatar
Michael Yang committed
40

41
42
43
    ```shell
    sudo yum install -y nvidia-container-toolkit
    ```
Michael Yang's avatar
Michael Yang committed
44

45
### Configure Docker to use Nvidia driver
46
47

```shell
Michael Yang's avatar
Michael Yang committed
48
49
50
51
sudo nvidia-ctk runtime configure --runtime=docker
sudo systemctl restart docker
```

52
### Start the container
Michael Yang's avatar
Michael Yang committed
53

54
```shell
Michael Yang's avatar
Michael Yang committed
55
56
57
docker run -d --gpus=all -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
```

58
59
60
61
<Note>
  If you're running on an NVIDIA JetPack system, Ollama can't automatically discover the correct JetPack version.
  Pass the environment variable `JETSON_JETPACK=5` or `JETSON_JETPACK=6` to the container to select version 5 or 6.
</Note>
62

63
## AMD GPU
Michael Yang's avatar
Michael Yang committed
64
65
66

To run Ollama using Docker with AMD GPUs, use the `rocm` tag and the following command:

67
```shell
Michael Yang's avatar
Michael Yang committed
68
69
70
docker run -d --device /dev/kfd --device /dev/dri -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama:rocm
```

71
## Run model locally
Michael Yang's avatar
Michael Yang committed
72
73
74

Now you can run a model:

75
```shell
76
docker exec -it ollama ollama run llama3.2
Michael Yang's avatar
Michael Yang committed
77
78
```

79
## Try different models
Michael Yang's avatar
Michael Yang committed
80
81

More models can be found on the [Ollama library](https://ollama.com/library).