fix zluda device_id

2026-09-19 01:04:32 +02:00 · 2024-09-01 11:21:08 +09:00
parent 2e02e73603
commit 66f06fba55
1 changed files with 1 additions and 0 deletions
@@ -510,6 +510,7 @@ def install_rocm_zluda():
        if error is None:
            if args.device_id is not None:
                os.environ['HIP_VISIBLE_DEVICES'] = args.device_id
+                del args.device_id
            try:
                zluda_installer.load(zluda_path)
                torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.3.0 torchvision --index-url https://download.pytorch.org/whl/cu118')