Comfy-Org / Comfy-Org/ComfyUI

ComfyUI needs Nvidia GPU even specifying ROCm

Open
#6,649 11 comments 4 reactions 0 assignees View on GitHub
Potential Bug
Dominant language
Python
Stars
133k
Forks
15.7k
Avg merge
1d 7h
Merged PRs (30d)
158

Description

### Expected Behavior

Can run ComfyUI on Linux using StabilityMatrix interface for managing installations with an AMD GPU

### Actual Behavior

Updating to the latest one in StabilityMatrix UI now has caused all versions of the installation to not run because now it requires Nvidia GPU, even though I specified ROCm during installation and get the following error:

`RuntimeError: Found no NVIDIA driver on your system.`

### Steps to Reproduce

1. Install StabilityMatrix
2. Add package for ComfyUI and specify ROCm and Symlink for the advanced installation options
3. Attemtp to run ComfyUI

### Debug Logs

```powershell
Checkpoint files will always be loaded safely.
Traceback (most recent call last):
File "/home/greg/Applications/StabilityMatrix/Packages/ComfyUI/main.py", line 136, in
import execution
File "/home/greg/Applications/StabilityMatrix/Packages/ComfyUI/execution.py", line 13, in
import nodes
File "/home/greg/Applications/StabilityMatrix/Packages/ComfyUI/nodes.py", line 22, in
import comfy.diffusers_load
File "/home/greg/Applications/StabilityMatrix/Packages/ComfyUI/comfy/diffusers_load.py", line 3, in
import comfy.sd
File "/home/greg/Applications/StabilityMatrix/Packages/ComfyUI/comfy/sd.py", line 6, in
from comfy import model_management
File "/home/greg/Applications/StabilityMatrix/Packages/ComfyUI/comfy/model_management.py", line 166, in
total_vram = get_total_memory(get_torch_device()) / (1024 * 1024)
File "/home/greg/Applications/StabilityMatrix/Packages/ComfyUI/comfy/model_management.py", line 129, in get_torch_device
return torch.device(torch.cuda.current_device())
File "/home/greg/Applications/StabilityMatrix/Packages/ComfyUI/venv/lib/python3.10/site-packages/torch/cuda/__init__.py", line 971, in current_device
_lazy_init()
File "/home/greg/Applications/StabilityMatrix/Packages/ComfyUI/venv/lib/python3.10/site-packages/torch/cuda/__init__.py", line 319, in _lazy_init
torch._C._cuda_init()
RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx
```

### Other

_No response_

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.