search

carpedm20 / multi-speaker-tacotron-tensorflow

Multi-speaker Tacotron in TensorFlow.
http://carpedm20.github.io/tacotron
Other
632 stars 341 forks source link

Running Training Code on GPU #45

Closed vijaysumaravi closed 6 years ago

vijaysumaravi commented 6 years ago

The training code runs fine on my CPU but how do I make it run on the GPUs? Thanks!

vijaysumaravi commented 6 years ago

sess_config = tf.ConfigProto( log_device_placement=False, allow_soft_placement=True, device_count = {'GPU': 1})