DmitryUlyanov/neural-style-audio-tf

Blank screen error

piteight opened this issue · 2 comments

Hi Dimitry :)
I have encountered some kind of error, when trying to transfer style from one song to another.
After running few cells, screen goes black and I cannot use keyboard nor mouse, cant enter tty mode - looks like regular system crash.
I'm using ubuntu 16.04 with tensorflow gpu (geforce 760gti 2gb vram).
Is this problem caused by using gpu version?

Hello @piteight, this code is cpu only as far as I remember, so it cannot be caused by gpu. Looks like something unrelated to this particular repo. Try reinstalling TF of ubuntu if nothing helps. Good luck!

@piteight
Did you fix your blank screen error? It is unrelated to gpu, but if you did want to try running on gpu, all you need to do is change the '/cpu:0' part of below line to '/gpu:0':

with g.as_default(), g.device('/cpu:0'), tf.Session() as sess: