Skip to content

Commit 22f4295

Browse files
authored
Update install.py for Intel Arc GPUs
update installation for Intel Arc GPUs utilizing upstream torch xpu. Comfy-Org/ComfyUI#7767
1 parent 1fd72a4 commit 22f4295

1 file changed

Lines changed: 7 additions & 11 deletions

File tree

comfy_cli/command/install.py

Lines changed: 7 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -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)