From 66f06fba55f1184ade67158fe2bf51e64c2f79a3 Mon Sep 17 00:00:00 2001
From: Seunghoon Lee <lshqqytiger@naver.com>
Date: Sun, 1 Sep 2024 11:21:08 +0900
Subject: [PATCH] fix zluda device_id

---
 installer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/installer.py b/installer.py
index cff0fd110..7ec09ae0a 100644
--- a/installer.py
+++ b/installer.py
@@ -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')