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

Dont detects cuda #134

Closed Paillat-dev closed 1 year ago

Paillat-dev commented 2 years ago

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

Originally posted by @Paillat-dev in https://github.com/lucidrains/lightweight-gan/issues/130#issuecomment-1209351237

terminatormlp commented 1 year ago

install pytorch gpu watch how to do there https://pytorch.org/get-started/locally/