Skip to content

Commit

Permalink
Merge pull request #146 from daronyondem/patch-1
Browse files Browse the repository at this point in the history
Adding --gpus=all to docker run command Fixes #121
  • Loading branch information
absadiki authored Sep 17, 2024
2 parents 6946886 + 2c2491c commit eda50ae
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ Simple examples can be found in the [examples](https://github.com/abdeladim-s/su
* Make sure that you have `docker` installed.
* Prebuilt image
1. ```docker pull absadiki/subsai:main```
2. ```docker run -p 8501:8501 -v /path/to/your/media_files/folder:/media_files absadiki/subsai:main```
2. ```docker run --gpus=all -p 8501:8501 -v /path/to/your/media_files/folder:/media_files absadiki/subsai:main```
* Build the image locally
1. Clone and `cd` to the repository
2. ```docker compose build```
Expand Down

0 comments on commit eda50ae

Please sign in to comment.