ComfyUI needs Nvidia GPU even specifying ROCm
- 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
Assessment
This issue has not been assessed yet.