diff --git a/README.md b/README.md index 57f0ede..6e54f41 100644 --- a/README.md +++ b/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 diff --git a/docker/README.md b/docker/README.md index 68646d0..9b2ee12 100644 --- a/docker/README.md +++ b/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.