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.