document gpu selection

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:
 ```bash
-python app.py
+CUDA_VISIBLE_DEVICES=0 python app.py
 ```
 
 ## Running with command line

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
 ```
 
+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.