It doesn't use my GPU? - Githubissues

tamarott / SinGAN

Official pytorch implementation of the paper: "SinGAN: Learning a Generative Model from a Single Natural Image"

https://tamarott.github.io/SinGAN.htm

Other

3.31k stars 611 forks source link

It doesn't use my GPU? #92

Open zg2358 opened 4 years ago

zg2358 commented 4 years ago

Hi, I have a GTX 1080Ti like they have stated in the supplementary material, but when I run commands like

python main_train.py --input_name colusseum.png

I notice that my GPU usage is almost 0%, but my CPU usage is much higher than idle. Is it supposed to behave like this, or do I need to reinstall some packages?