Expand cupy install guide in README
Add step-by-step instructions, CUDA version table, troubleshooting section, and note that EMA-VFI works without cupy. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
38
README.md
38
README.md
@@ -159,23 +159,41 @@ pip install -r requirements.txt
|
|||||||
|
|
||||||
### cupy (required for BIM-VFI, SGM-VFI, GIMM-VFI)
|
### cupy (required for BIM-VFI, SGM-VFI, GIMM-VFI)
|
||||||
|
|
||||||
`cupy` is needed for optical flow warping in BIM-VFI, SGM-VFI, and GIMM-VFI. EMA-VFI does **not** need cupy.
|
[cupy](https://cupy.dev/) is a GPU-accelerated array library used for optical flow warping. It is required by **BIM-VFI**, **SGM-VFI**, and **GIMM-VFI**. EMA-VFI does **not** need cupy and works without it.
|
||||||
|
|
||||||
If cupy is missing, the Load node will show an error in ComfyUI with your CUDA version and the exact install command. To install:
|
cupy must match your PyTorch CUDA version. If it is missing or mismatched, the Load node will show an error in ComfyUI with your CUDA version and the exact install command.
|
||||||
|
|
||||||
|
#### How to install cupy
|
||||||
|
|
||||||
|
**Step 1 — Find your CUDA version:**
|
||||||
|
|
||||||
1. Find your CUDA version:
|
|
||||||
```bash
|
```bash
|
||||||
python -c "import torch; print(torch.version.cuda)"
|
python -c "import torch; print(torch.version.cuda)"
|
||||||
```
|
```
|
||||||
|
|
||||||
2. Install the matching cupy package:
|
This prints something like `12.4` or `11.8`.
|
||||||
```bash
|
|
||||||
# CUDA 12.x
|
|
||||||
pip install cupy-cuda12x
|
|
||||||
|
|
||||||
# CUDA 11.x
|
**Step 2 — Install the matching cupy package:**
|
||||||
pip install cupy-cuda11x
|
|
||||||
```
|
| CUDA version | Install command |
|
||||||
|
|---|---|
|
||||||
|
| 12.x | `pip install cupy-cuda12x` |
|
||||||
|
| 11.x | `pip install cupy-cuda11x` |
|
||||||
|
|
||||||
|
> **Note:** Make sure to run pip in the same Python environment as ComfyUI. If you use a venv or conda, activate it first.
|
||||||
|
|
||||||
|
#### Troubleshooting
|
||||||
|
|
||||||
|
| Problem | Solution |
|
||||||
|
|---------|----------|
|
||||||
|
| `ModuleNotFoundError: No module named 'cupy'` | Install cupy using the steps above |
|
||||||
|
| `cupy` installed but `ImportError` at runtime | CUDA version mismatch — uninstall with `pip uninstall cupy-cuda12x` and reinstall the correct version |
|
||||||
|
| Install hangs or takes very long | cupy wheels are large (~800MB). Use a fast connection and be patient |
|
||||||
|
| Docker / no build tools | Use the prebuilt wheel: `pip install cupy-cuda12x` (not `cupy` which compiles from source) |
|
||||||
|
|
||||||
|
#### Can I skip cupy?
|
||||||
|
|
||||||
|
Yes — just use **EMA-VFI**, which does not require cupy. It is the fastest model and uses the least VRAM. The other three models (BIM-VFI, SGM-VFI, GIMM-VFI) will not load without cupy.
|
||||||
|
|
||||||
### Other dependencies
|
### Other dependencies
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user