How to train with GPU on my own pc?

[drizzt00s]

How to train with GPU on my own pc? Can anyone give a brief introduction?

[prasys]

Basically you install anaconda , install gpt-2-simple and follow the instructions. pretty straight forward

i am running it on my local machine :)

[drizzt00s]

thx bro, I give it up. too many strange oom problems occurred. I have only a 1550ti and switch back to cpu

[binkiebinkie]

I don't know how to switch between GPU and CPU, but you can check what you're using by logging with something similar to the following: sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) then in the output look for your GPU's name https://stackoverflow.com/questions/38009682/how-to-tell-if-tensorflow-is-using-gpu-acceleration-from-inside-python-shell