Update README.md
Browse files
README.md
CHANGED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
Introduction to Installing Custom PyTorch Wheels for ComfyUI
|
| 2 |
+
This guide outlines the process of installing pre-built PyTorch wheels for use within a ComfyUI virtual environment. These wheels are specifically compiled for Python 3.11, CUDA 12.4, and are designed to enhance performance and compatibility.
|
| 3 |
+
|
| 4 |
+
Installation Steps:
|
| 5 |
+
|
| 6 |
+
Activate your ComfyUI virtual environment. This ensures that the custom PyTorch installation is isolated to your ComfyUI setup and avoids conflicts with other Python environments.
|
| 7 |
+
|
| 8 |
+
Bash
|
| 9 |
+
E:\ComfyUI>venv\Scripts\activate
|
| 10 |
+
|
| 11 |
+
|
| 12 |
+
Install the custom PyTorch wheel. Assuming you have downloaded the torch-2.5.1+cu124-cp311-cp311-win_amd64.whl file to the E: drive, use the following pip command:
|
| 13 |
+
|
| 14 |
+
https://download.pytorch.org/whl/cu124/torch-2.5.1%2Bcu124-cp311-cp311-win_amd64.whl#sha256=6c8a7003ef1327479ede284b6e5ab3527d3900c2b2d401af15bcc50f2245a59f
|
| 15 |
+
Bash
|
| 16 |
+
pip install "E:\torch-2.5.1+cu124-cp311-cp311-win_amd64.whl"
|
| 17 |
+
|
| 18 |
+
|
| 19 |
+
This command instructs pip to install the specified wheel file. The quotation marks are used to handle potential spaces in the file path.
|
| 20 |
+
Install the corresponding Torchvision and Torchaudio extensions. To ensure compatibility with the installed PyTorch version, install the matching Torchvision and Torchaudio extensions using the following command:
|
| 21 |
+
|
| 22 |
+
Bash
|
| 23 |
+
pip install torchvision==0.20.1+cu124 torchaudio==2.5.1+cu124 --index-url https://download.pytorch.org/whl/cu124
|
| 24 |
+
|
| 25 |
+
This command installs specific versions of Torchvision and Torchaudio that are compatible with CUDA 12.4. The --index-url flag specifies the PyTorch wheel index, ensuring that the correct CUDA-enabled versions are downloaded.
|
| 26 |
+
Key Considerations:
|
| 27 |
+
Virtual Environment: It is crucial to perform these installations within your ComfyUI virtual environment to maintain a clean and organized setup.
|
| 28 |
+
File Path: Ensure that the file path to your downloaded PyTorch wheel is correct.
|
| 29 |
+
CUDA Compatibility: These wheels are specifically built for CUDA 12.4. If you have a different CUDA version, you will need to obtain or build compatible wheels.
|
| 30 |
+
Version Matching: Always ensure that the versions of PyTorch, Torchvision, and Torchaudio are compatible to avoid potential errors.
|
| 31 |
+
By following these steps, you can successfully install custom PyTorch wheels and their extensions within your ComfyUI environment, enabling optimized performance for your workflows.
|