Browse Source

document gpu selection

main
catid 8 months ago
parent
commit
3fb546398c
  1. 2
      README.md
  2. 6
      docker/README.md

2
README.md

@ -68,7 +68,7 @@ By default, we use the `instant-mesh-large` reconstruction model variant.
To start a gradio demo in your local machine, simply running: To start a gradio demo in your local machine, simply running:
```bash ```bash
python app.py CUDA_VISIBLE_DEVICES=0 python app.py
``` ```
## Running with command line ## Running with command line

6
docker/README.md

@ -19,4 +19,10 @@ export MODEL_DIR=$HOME/models/
docker run -it -p 43839:43839 --platform=linux/amd64 --gpus all -v $MODEL_DIR:/workspace/instantmesh/models instantmesh docker run -it -p 43839:43839 --platform=linux/amd64 --gpus all -v $MODEL_DIR:/workspace/instantmesh/models instantmesh
``` ```
To use specific GPUs:
```bash
docker run -it -p 43839:43839 --platform=linux/amd64 --gpus '"device=0,1"' -v $MODEL_DIR:/workspace/instantmesh/models instantmesh
```
Navigate to `http://localhost:43839` to use the demo. Navigate to `http://localhost:43839` to use the demo.

Loading…
Cancel
Save