Skip to content

Commit 4de0975

Browse files
authored
Update install.py for Intel Arc GPUs (#319)
1 parent 1fd72a4 commit 4de0975

File tree

1 file changed

+8
-12
lines changed

1 file changed

+8
-12
lines changed

comfy_cli/command/install.py

Lines changed: 8 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from rich.panel import Panel
1414
from rich.prompt import Confirm
1515

16-
from comfy_cli import constants, ui, utils
16+
from comfy_cli import constants, ui
1717
from comfy_cli.command.custom_nodes.command import update_node_id_cache
1818
from comfy_cli.command.github.pr_info import PRInfo
1919
from comfy_cli.constants import GPU_OPTION
@@ -96,28 +96,24 @@ def pip_install_comfyui_dependencies(
9696
base_command,
9797
check=False,
9898
)
99-
# Beta support for intel arch based on this PR: https://github.com/comfyanonymous/ComfyUI/pull/3439
99+
# Update installation to use upstream torch xpu. ipex is no longer needed for Intel Arc GPUs
100+
# https://github.com/comfyanonymous/ComfyUI/pull/7767
100101
if gpu == GPU_OPTION.INTEL_ARC:
101102
pip_url = [
102103
"--extra-index-url",
103-
"https://pytorch-extension.intel.com/release-whl/stable/xpu/us/",
104+
"https://download.pytorch.org/whl/xpu",
104105
]
105-
utils.install_conda_package("libuv")
106+
106107
# TODO: wrap pip install in a function
107-
subprocess.run(
108-
[sys.executable, "-m", "pip", "install", "mkl", "mkl-dpcpp"],
109-
check=True,
110-
)
111108
result = subprocess.run(
112109
[
113110
sys.executable,
114111
"-m",
115112
"pip",
116113
"install",
117-
"torch==2.1.0.post2",
118-
"torchvision==0.16.0.post2",
119-
"torchaudio==2.1.0.post2",
120-
"intel-extension-for-pytorch==2.1.30",
114+
"torch",
115+
"torchvision",
116+
"torchaudio",
121117
]
122118
+ pip_url,
123119
check=False,

0 commit comments

Comments
 (0)