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

tensorflow 导入新的tensorflow实例相关推荐

python 调用 tensorflow.dll_python tensorflow导入DLL加载失败
我安装了最新的python 3.6.4 x64版本然后使用pip3为cpu-only安装tensorflow C:\>pip3 install tensorflow 但是当我尝试在python ...
tensorflow导入自己的数据集
在构建tensorflow模型过程中,可谓是曲折颇多,一些教程上教会了我们如何使用下载的现成数据集,但却没有提及如何构建自己的数据集.我自己在学习过程中也走了不少弯路,希望这一系列的博客能解决大家的一 ...
python加载dll失败_python tensorflow导入DLL加载失败
我安装了最新的 python 3.6.4 x64版本然后使用pip3为cpu-only安装tensorflow C:\>pip3 install tensorflow 但是当我尝试在pytho ...
tensorflow学习笔记——使用TensorFlow操作MNIST数据（1）
续集请点击我:tensorflow学习笔记--使用TensorFlow操作MNIST数据(2) 本节开始学习使用tensorflow教程,当然从最简单的MNIST开始.这怎么说呢,就好比编程入门有He ...
TensorFlow Hub介绍：TensorFlow中可重用的机器学习模块库
摘要: 本文对TensorFlow Hub库的介绍,并举例说明其用法. 在软件开发中,最常见的失误就是容易忽视共享代码库,而库则能够使软件开发具有更高的效率.从某种意义上来说,它改变了编程的过程.我们 ...
谷歌发布TensorFlow 1.4与TensorFlow Lattice：利用先验知识提升模型准确度搜狐科技 10-12 15:29 选自：Google Research Blog 参与：李泽南、
谷歌发布TensorFlow 1.4与TensorFlow Lattice:利用先验知识提升模型准确度昨天,谷歌发布了 TensorFlow 1.4.0 先行版,将 tf.data 等功能加入了 A ...
DL框架之TensorFlow：深度学习框架TensorFlow Core(低级别TensorFlow API)的简介、安装、使用方法之详细攻略
DL框架之TensorFlow:TensorFlow Core(低级别TensorFlow API)的简介.安装.使用方法之详细DL框架之TensorFlow:深度学习框架TensorFlow Cor ...
8.2 TensorFlow实现KNN与TensorFlow中的损失函数,优化函数
前言 8.1 mnist_soft,TensorFlow构建回归模型中对主要对计算图的概念与公式与计算图的转化进行了介绍,8.2则主要介绍一下TensorFlow中自带的几个算子,与优化函数,损失函数 ...
【TensorFlow】：解决TensorFlow的ImportError: DLL load failed: 动态链接库(DLL)初始化例程失败...
[背景] 在scikit-learn基础上系统结合数学和编程的角度学习了机器学习后(我的github:https://github.com/wwcom614/machine-learning),意犹未 ...

tensorflow 导入新的tensorflow实例

tensorflow 导入新的tensorflow实例相关推荐

最新文章

热门文章