PIE二次开发,调用tensorflow-gpu时出现的问题

PIE二次开发,调用tensorflow-gpu时出现的问题

0、运行环境

  1. VS2015,
  2. PIE32位,
  3. python3.7,
  4. tensorflow-gpu==2.0.0

1、问题

winform应用窗体调用tensorflow搭建的算法时,出现假死现象。
已成功调用算法文件,但是算法不往下执行。
使用CMD直接执行算法是没有问题的。

2、解决思路

切换为tensorflow-cpu版本时,上述问题就不会出现
但是我不能用CPU去训练模型呀
这时,在CMD中对比CPU版本和GPU版本执行算法时的差异性,
观察到,在执行tensorflow-GPU版本的算法时,控制台会打印很多CUDA、CUDNN等通知信息和警告信息。
猜想:是不是这些信息winform应用窗体无法获取呢?
这时想着在代码中取消这些信息的打印,
参考了(这个链接是转载)【tensorflow-2.x-gpu 】 屏蔽tensorflow输出的log信息

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

引入这行代码,且放在import tensorflow as tf之前,即可关闭无用的打印输出,
此时在winform应用窗体调用tensorflow算法时,就可以正常执行了。

注意:

这个是用在tensorflow2.x版本的,在tensorflow1.x版本中是否适用暂且不知。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值