Skip to content

Commit 6f0c13f

Browse files
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

File tree

gpustack_runtime/envs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -381,7 +381,7 @@
381381
"GPUSTACK_RUNTIME_DEPLOY_RESOURCE_KEY_MAP_BACKEND_VISIBLE_DEVICES": lambda: to_dict(
382382
getenv(
383383
"GPUSTACK_RUNTIME_DEPLOY_RESOURCE_KEY_MAP_BACKEND_VISIBLE_DEVICES",
384-
"amd.com/devices=HIP_VISIBLE_DEVICES,ROCR_VISIBLE_DEVICES;"
384+
"amd.com/devices=HIP_VISIBLE_DEVICES;"
385385
"huawei.com/devices=ASCEND_RT_VISIBLE_DEVICES,NPU_VISIBLE_DEVICES;"
386386
"cambricon.com/devices=MLU_VISIBLE_DEVICES;"
387387
"hygon.com/devices=HIP_VISIBLE_DEVICES;"

0 commit comments

Comments
 (0)