Commit 6f0c13f
committed
fix(amd): failed to assign specific device
Misconfiguration of `HIP_VISIBLE_DEVICES` and `ROCR_VISIBLE_DEVICES`,
according to pytorch/pytorch#140318,
use `HIP_VISIBLE_DEVICES` is enough.
Signed-off-by: thxCode <thxcode0824@gmail.com>1 parent 44deb12 commit 6f0c13f
1 file changed
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
381 | 381 | | |
382 | 382 | | |
383 | 383 | | |
384 | | - | |
| 384 | + | |
385 | 385 | | |
386 | 386 | | |
387 | 387 | | |
| |||
0 commit comments