mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
[Bug] fix bug: delete & (#1365)
Co-authored-by: 白超 <baichao19@huawei.com>
This commit is contained in:
parent
69aa2f2d57
commit
51a94aee01
@ -25,7 +25,7 @@ def get_command_template(gpu_ids: List[int]) -> str:
|
||||
"""Format command template given available gpu ids."""
|
||||
if is_npu_available():
|
||||
tmpl = 'ASCEND_RT_VISIBLE_DEVICES=' + ','.join(str(i) for i in gpu_ids)
|
||||
tmpl += ' & {task_cmd}'
|
||||
tmpl += ' {task_cmd}'
|
||||
elif sys.platform == 'win32': # Always return win32 for Windows
|
||||
# use command in Windows format
|
||||
tmpl = 'set CUDA_VISIBLE_DEVICES=' + ','.join(str(i) for i in gpu_ids)
|
||||
|
Loading…
Reference in New Issue
Block a user