Found no NVIDIA driver on your system

#1
by nightfury - opened

Dear all, I'm facing following Error in every Build.. kindly help with the space..

File "app.py", line 22, in
img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
File "/home/user/app/src/diffusers/src/diffusers/pipeline_utils.py", line 383, in from_pretrained
loaded_sub_model = load_method(os.path.join(cached_folder, name), **loading_kwargs)
File "/home/user/app/src/diffusers/src/diffusers/configuration_utils.py", line 160, in from_config
model = cls(**init_dict)
File "/home/user/app/src/diffusers/src/diffusers/configuration_utils.py", line 382, in inner_init
init(self, *args, **init_kwargs)
File "/home/user/app/src/diffusers/src/diffusers/schedulers/scheduling_pndm.py", line 131, in init
self.set_format(tensor_format=tensor_format)
File "/home/user/app/src/diffusers/src/diffusers/schedulers/scheduling_utils.py", line 53, in set_format
setattr(self, key, torch.from_numpy(value).to(device='cuda:0'))
File "/home/user/.local/lib/python3.8/site-packages/torch/cuda/init.py", line 217, 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

Sign up or log in to comment