No NVIDIA in _lazy_init

#9
by AmirPiran - opened

I have an AMD GPU and the model requires an NVIDIA GPU. The error I got is :
File "/venv/lib/python3.10/site-packages/torch/cuda/init.py", line 298, 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

I can't also find the file and change it. Can you help me?

Sign up or log in to comment