Skip to content

Commit 211facf

Browse files
authored
Merge pull request #6 from LykosAI/deps-update
Add cuda 12 wheels for 3.11, 3.12, and Linux
2 parents b996c18 + 8c519c4 commit 211facf

File tree

2 files changed

+35
-1
lines changed

2 files changed

+35
-1
lines changed

README.md

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,25 @@
11
# ComfyUI-Inference-Core-Nodes
22

3+
## Installation
4+
1. [Stability Matrix](https://github.com/LykosAI/StabilityMatrix) Extensions Manager
5+
2. [ComfyUI Manager](https://github.com/ltdrdata/ComfyUI-Manager)
6+
7+
## Manual Installation
8+
1. Clone this repository to `ComfyUI/custom_nodes/`
9+
10+
2. Either:
11+
- Run `install.py` using the venv or preferred python environment.
12+
13+
Or
14+
15+
(Installs required dependencies and appropriate onnxruntime acceleration via compiled wheels)
16+
- (CUDA 11 or latest stable) Run `pip install -e .[cuda]`
17+
- (CUDA 12) Run `pip install -e .[cuda12]`
18+
- (RoCM) Run `pip install -e .[rocm]`
19+
- (DirectML) Run `pip install -e .[dml]`
20+
- (CPU Only) Run `pip install -e .[cpu]`
21+
22+
Or
23+
24+
(Installs only required dependencies without onnxruntime acceleration)
25+
- Run `pip install -e .`

pyproject.toml

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,18 @@ cuda = [
3333
"onnxruntime-gpu"
3434
]
3535
cuda-12 = [
36-
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/2692857e-05ef-43b4-ba9c-ccf1c22c437c/_packaging/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/download/onnxruntime-gpu/1.17.1/onnxruntime_gpu-1.17.1-cp310-cp310-win_amd64.whl ; platform_system == 'Windows'"
36+
# CUDA 12, Python 3.10, Windows
37+
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/PublicPackages/_apis/packaging/feeds/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/packages/onnxruntime-gpu/versions/1.17.1/onnxruntime_gpu-1.17.1-cp310-cp310-win_amd64.whl/content ; platform_system == 'Windows' and python_version == '3.10'",
38+
# CUDA 12, Python 3.10, Linux
39+
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/PublicPackages/_apis/packaging/feeds/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/packages/onnxruntime-gpu/versions/1.17.1/onnxruntime_gpu-1.17.1-cp311-cp311-manylinux_2_28_x86_64.whl/content ; platform_system == 'Linux' and python_version == '3.10'",
40+
# CUDA 12, Python 3.11, Windows
41+
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/PublicPackages/_apis/packaging/feeds/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/packages/onnxruntime-gpu/versions/1.17.1/onnxruntime_gpu-1.17.1-cp311-cp311-win_amd64.whl/content ; platform_system == 'Windows' and python_version == '3.11'",
42+
# CUDA 12, Python 3.11, Linux
43+
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/PublicPackages/_apis/packaging/feeds/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/packages/onnxruntime-gpu/versions/1.17.1/onnxruntime_gpu-1.17.1-cp311-cp311-manylinux_2_28_x86_64.whl/content ; platform_system == 'Linux' and python_version == '3.11'",
44+
# CUDA 12, Python 3.12, Windows
45+
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/PublicPackages/_apis/packaging/feeds/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/packages/onnxruntime-gpu/versions/1.17.1/onnxruntime_gpu-1.17.1-cp312-cp312-win_amd64.whl/content ; platform_system == 'Windows' and python_version == '3.12'",
46+
# CUDA 12, Python 3.12, Linux
47+
"onnxruntime-gpu @ https://aiinfra.pkgs.visualstudio.com/PublicPackages/_apis/packaging/feeds/9387c3aa-d9ad-4513-968c-383f6f7f53b8/pypi/packages/onnxruntime-gpu/versions/1.17.1/onnxruntime_gpu-1.17.1-cp312-cp312-manylinux_2_28_x86_64.whl/content ; platform_system == 'Linux' and python_version == '3.12'"
3748
]
3849
directml = [
3950
"onnxruntime-directml"

0 commit comments

Comments
 (0)