[Bug] fix bug: delete & (#1365)

Co-authored-by: 白超 <baichao19@huawei.com>
This commit is contained in:
baymax591 2024-07-26 11:03:55 +08:00 committed by GitHub
parent 69aa2f2d57
commit 51a94aee01
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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)