Skip to content

Commit 88e9ae6

Browse files
committed
fix
1 parent 72f1fe4 commit 88e9ae6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modules/launch_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -534,8 +534,8 @@ def prepare_environment():
534534
if args.use_rocm or args.use_zluda:
535535
print('No ROCm agent was found. Please make sure that graphics driver is installed and up to date.')
536536
backend = "cpu"
537+
args.skip_torch_cuda_test = True
537538
else:
538-
print('ROCm: AMD toolkit detected')
539539
print(f'ROCm: agents={[gpu.name for gpu in amd_gpus]}')
540540
if args.device_id is None:
541541
index = 0

0 commit comments

Comments
 (0)