From 38959f89dc7264dacd541061d1c9d9e56d2d07a2 Mon Sep 17 00:00:00 2001 From: su Date: Wed, 6 Mar 2024 21:12:34 -0500 Subject: [PATCH] docker cpu only #120 --- README.md | 4 ++-- docker-compose.yml | 7 ++++++- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 53da340..2bf3f64 100644 --- a/README.md +++ b/README.md @@ -191,8 +191,8 @@ Simple examples can be found in the [examples](https://github.com/abdeladim-s/su 1. Make sure that you have `docker` installed. 2. Clone and `cd` to the repository 3. ```docker compose build``` -4. ```docker compose run -p 8501:8501 -v /path/to/your/media_files/folder:/media_files subsai-webui``` -5. You can access your media files through the mounted `media_files` folder. +4. ```docker compose run -p 8501:8501 -v /path/to/your/media_files/folder:/media_files subsai-webui # subsai-webui-cpu for cpu only``` +6. You can access your media files through the mounted `media_files` folder. # Notes * If you have an NVIDIA graphics card, you may need to install [cuda](https://docs.nvidia.com/cuda/#installation-guides) to use the GPU capabilities. diff --git a/docker-compose.yml b/docker-compose.yml index c3a88bb..2a134c2 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -10,4 +10,9 @@ services: reservations: devices: - driver: nvidia - capabilities: [ gpu ] \ No newline at end of file + capabilities: [ gpu ] + + subsai-webui-cpu: + build: + context: . + dockerfile: Dockerfile \ No newline at end of file