Skip to content

Commit

Permalink
docs: document UUID support
Browse files Browse the repository at this point in the history
  • Loading branch information
JoanFM committed Nov 7, 2022
1 parent b500e09 commit 3f6ed46
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion docs/fundamentals/flow/topologies.md
Original file line number Diff line number Diff line change
Expand Up @@ -147,7 +147,7 @@ You can restrict the visible devices in round-robin assignment using `CUDA_VISIB
| 0 | 4 |


You can restrict the visible devices in round-robin assignment by assigning a list of devices ids `CUDA_VISIBLE_DEVICES=RR1,3`. This creates the following assignment:
You can restrict the visible devices in round-robin assignment by assigning a list of devices IDS `CUDA_VISIBLE_DEVICES=RR1,3`. This creates the following assignment:

| GPU device | Replica ID |
|------------|------------|
Expand All @@ -157,6 +157,15 @@ You can restrict the visible devices in round-robin assignment by assigning a li
| 3 | 3 |
| 1 | 4 |

If your [CUDA driver](https://docs.nvidia.com/datacenter/tesla/mig-user-guide/index.html#cuda-baremetal) allows to, you can also refer to GPUs by their UUID. For instance, you could assign a list of device UUIDs `CUDA_VISIBLE_DEVICES=RRGPU-0aaaaaaa-74d2-7297-d557-12771b6a79d5,GPU-0bbbbbbb-74d2-7297-d557-12771b6a79d5,GPU-0ccccccc-74d2-7297-d557-12771b6a79d5,GPU-0ddddddd-74d2-7297-d557-12771b6a79d5`

| GPU device | Replica ID |
|------------|------------|
| GPU-0aaaaaaa-74d2-7297-d557-12771b6a79d5 | 0 |
| GPU-0bbbbbbb-74d2-7297-d557-12771b6a79d5 | 1 |
| GPU-0ccccccc-74d2-7297-d557-12771b6a79d5 | 2 |
| GPU-0ddddddd-74d2-7297-d557-12771b6a79d5 | 3 |
| GPU-0aaaaaaa-74d2-7297-d557-12771b6a79d5 | 4 |

## Distributed replicas

Expand Down

0 comments on commit 3f6ed46

Please sign in to comment.