PIE二次开发,调用tensorflow-gpu时出现的问题
0、运行环境
- VS2015,
- PIE32位,
- python3.7,
- 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版本中是否适用暂且不知。