From 075dddf3d3e8492b8ff24c91d002065b73a1f015 Mon Sep 17 00:00:00 2001 From: Justin Hayes Date: Fri, 5 Apr 2024 20:11:33 -0400 Subject: [PATCH 1/3] Add AMD docker compose file --- docker-compose.amdgpu.yaml | 8 ++++++++ 1 file changed, 8 insertions(+) create mode 100644 docker-compose.amdgpu.yaml diff --git a/docker-compose.amdgpu.yaml b/docker-compose.amdgpu.yaml new file mode 100644 index 00000000..130b1ccb --- /dev/null +++ b/docker-compose.amdgpu.yaml @@ -0,0 +1,8 @@ +services: + ollama: + devices: + - /dev/kfd:/dev/kfd + - /dev/dri:/dev/dri + image: ollama/ollama:rocm + environment: + - 'HSA_OVERRIDE_GFX_VERSION=${HSA_OVERRIDE_GFX_VERSION}' \ No newline at end of file From 8db03f3ab28b2ae98173352dafc189d0b7cd9837 Mon Sep 17 00:00:00 2001 From: Justin Hayes Date: Fri, 5 Apr 2024 23:46:20 -0400 Subject: [PATCH 2/3] Add variables --- docker-compose.amdgpu.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker-compose.amdgpu.yaml b/docker-compose.amdgpu.yaml index 130b1ccb..7a1295d9 100644 --- a/docker-compose.amdgpu.yaml +++ b/docker-compose.amdgpu.yaml @@ -3,6 +3,6 @@ services: devices: - /dev/kfd:/dev/kfd - /dev/dri:/dev/dri - image: ollama/ollama:rocm + image: ollama/ollama:${OLLAMA_DOCKER_TAG-rocm} environment: - - 'HSA_OVERRIDE_GFX_VERSION=${HSA_OVERRIDE_GFX_VERSION}' \ No newline at end of file + - 'HSA_OVERRIDE_GFX_VERSION=${HSA_OVERRIDE_GFX_VERSION-11.0.0}' \ No newline at end of file From f34b9733e33ea6f525c3733c96720d45fe749535 Mon Sep 17 00:00:00 2001 From: Justin Hayes Date: Fri, 5 Apr 2024 23:48:16 -0400 Subject: [PATCH 3/3] Add Docker tag variables --- docker-compose.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker-compose.yaml b/docker-compose.yaml index f69084b8..9daba312 100644 --- a/docker-compose.yaml +++ b/docker-compose.yaml @@ -8,7 +8,7 @@ services: pull_policy: always tty: true restart: unless-stopped - image: ollama/ollama:latest + image: ollama/ollama:${OLLAMA_DOCKER_TAG-latest} open-webui: build: @@ -16,7 +16,7 @@ services: args: OLLAMA_BASE_URL: '/ollama' dockerfile: Dockerfile - image: ghcr.io/open-webui/open-webui:main + image: ghcr.io/open-webui/open-webui:${WEBUI_DOCKER_TAG-main} container_name: open-webui volumes: - open-webui:/app/backend/data