Can this be trained using GPU?

yu4u / noise2noise

An unofficial and partial Keras implementation of "Noise2Noise: Learning Image Restoration without Clean Data"

MIT License

1.07k stars 237 forks source link

Can this be trained using GPU? #35

Open qazwsdc opened 5 years ago

qazwsdc commented 5 years ago

It's very slow using CPU to train a model

yu4u commented 5 years ago

You can train the model using a GPU simply by following README.

qazwsdc commented 5 years ago

I looked closely but didn't find any information about GPU.

qazwsdc commented 5 years ago

Could you please give some suggestions? Thank you

yu4u commented 5 years ago

Keras automatically uses GPU if it is available.

qazwsdc commented 5 years ago

I use RTX 2080ti .I have followed your README step but it's totally not use GPU at all. Maybe should I install some cuda drivers? My system is Windows 10.

yu4u commented 5 years ago

You might have to install tensorflow-gpu. Anyway, I think this is not the problem of this repository, but the environment of the Keras framework. Thus please refer to different documentations for appropriate environment construction.