CPU only
Nvidia GPU
Install the NVIDIA Container Toolkit.Install with Apt
-
Configure the repository
-
Install the NVIDIA Container Toolkit packages
Install with Yum or Dnf
-
Configure the repository
-
Install the NVIDIA Container Toolkit packages
Configure Docker to use Nvidia driver
Start the container
[!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.
AMD GPU
To run Ollama using Docker with AMD GPUs, use therocm
tag and the following command: