frozenleaves/DeepSpeed
1
0
Fork 0
mirror of https://github.com/deepspeedai/DeepSpeed.git synced 2025-10-20 15:33:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix an outdated doc wrt CUDA_VISIBLE_DEVICES (#7058)

Browse Source 
@jeffra and I fixed this many years ago, so bringing this doc to a
correct state.

---------

Signed-off-by: Stas Bekman <stas@stason.org>
This commit is contained in:
Stas Bekman
2025-02-20 07:27:54 -08:00
committed by GitHub
parent fa8967e209
commit 461d641f00
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
docs/_tutorials/getting-started.md
View File

@ -316,10 +316,14 @@ local machine to discover the number of slots available. The `--include` and
`--exclude` arguments work as normal, but the user should specify 'localhost'
as the hostname.
Also note that `CUDA_VISIBLE_DEVICES` can't be used with DeepSpeed to control
which devices should be used. For example, to use only gpu1 of the current
node, do:
Also note that `CUDA_VISIBLE_DEVICES` can be used with `deepspeed` to control
which devices should be used on a single node. So either of these would work
to launch just on devices 0 and 1 of the current node:
```bash
deepspeed --include localhost:1 ...
deepspeed --include localhost:0,1 ...
```
```bash
CUDA_VISIBLE_DEVICES=0,1 deepspeed ...
```