GPU problem

[Anooyman]

Hi takerum, Thanks for your work. In your code, it seems that you use the CPU to deal with the dataset and then use GPU to build the training graph. However, when I ran this program, here comes an issue that the program can not locate the CPU and GPU. Is there something I missed?

[wzm2256]

Did you solve this problem?

[wzm2256]

I solved it like this:

Add a line: config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=False)

and replace the session definition: with sv.managed_session() as sess: with: with sv.managed_session(config=config) as sess:

[Anooyman]

Thanks！！That helps a lot!

------------------ 原始邮件 ------------------ 发件人: "wzm2256"[email protected]; 发送时间: 2019年9月17日(星期二) 下午3:14 收件人: "takerum/vat_tf"[email protected]; 抄送: "Anooyman"[email protected];"Author"[email protected]; 主题: Re: [takerum/vat_tf] GPU problem (#16)

I solved it like this:

Add a line: config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=False)

and replace the session definition: with sv.managed_session() as sess: to be: with sv.managed_session(config=config) as sess:

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or mute the thread.