Doesn't detect CUDA from Nvidia GPU

lucidrains / lightweight-gan

Implementation of 'lightweight' GAN, proposed in ICLR 2021, in Pytorch. High resolution image generations that can be trained within a day or two

MIT License

1.62k stars 220 forks source link

Closed demiahmed closed 2 years ago

demiahmed commented 2 years ago

Error when trying to run lightweight_gan from the command line, I am getting the error

You need to have an Nvidia GPU with CUDA installed

Although running torch.cuda.is_available() returns True

Paillat-dev commented 2 years ago

Hello, I have the same problem, but for me it returns false, while I have already installed cuda.