tensorflow 导入新的tensorflow实例

因为涉及到同一台电脑多个GPU，在指定tensorflow图的时候，需要为不同的图指定不同的GPU，所以必须在导入tensorflow之前，指定可用的GPU

def import_tf(device_id=-1, verbose=False):"""import tensorflow, set tensorflow graph load device, set tensorflow log level, return tensorflow instance:param device_id: GPU id:param verbose: tensorflow logging level:return: tensorflow instance"""# set visible gpu, -1 is cpuos.environ['CUDA_VISIBLE_DEVICES'] = '-1' if device_id < 0 else str(device_id)os.environ['TF_CPP_MIN_LOG_LEVEL'] = '0' if verbose else '3'import tensorflow as tftf.logging.set_verbosity(tf.logging.DEBUG if verbose else tf.logging.ERROR)return tf

 

转载于:https://www.cnblogs.com/callyblog/p/10388454.html

《新程序员》：云原生和全面数字化实践50位技术专家共同创作，文字、视频、音频交互阅读

总结

以上是生活随笔为你收集整理的tensorflow 导入新的tensorflow实例的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。