doang commited on
Commit
feaa5b1
·
verified ·
1 Parent(s): 76dffff

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +31 -0
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.