kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

Updated docker run gpu

Browse Source
This commit is contained in:
loretoparisi 2021-11-15 23:24:07 +01:00
parent e839a36503
commit c8e3604f71
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -27,7 +27,7 @@ docker run -p 8080:8080 -e cache_dir=/app/models -v models:/app/models -v $(pwd)
## Docker (gpu)
```
docker build -f Dockerfile -t lamacleaner .
docker run --gpus all -p 8080:8080 -e cache_dir=/app/models -v models:/app/models -v $(pwd):/app --rm lamacleaner python3 main.py --device=gpu --port=8080
docker run --gpus all -p 8080:8080 -e cache_dir=/app/models -v models:/app/models -v $(pwd):/app --rm lamacleaner python3 main.py --device=cuda --port=8080
```
Then open [http://localhost:8080](http://localhost:8080)