tensorflow
有石为玉
个性化推荐,图像与视频处理,机器学习,深度学习
展开
-
linux ubuntu16.4.0 anaconda下安装tensorflow1.10.0的cpu/gpu版本踩坑记录
1、安装tensorflow-gpu后,import出现错误:ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directorytf版本:1.13.1,cuda版本:9.0,不匹配造成的。看下tf和cuda版本对照:https://blog.csdn.net/omodao...原创 2019-05-08 14:32:26 · 2044 阅读 · 0 评论 -
linux判断安装的tensorflow是gpu还是cpu版本
判断代码:import osfrom tensorflow.python.client import device_libos.environ["TF_CPP_MIN_LOG_LEVEL"] = "99" if __name__ == "__main__": print(device_lib.list_local_devices())结果一:CPU[name: "/...原创 2019-07-12 15:21:27 · 1751 阅读 · 0 评论 -
keras转tensorflow后,报错:Attempting to use uninitialized value batch_normalization_14/moving_mean
keras转tensorflow,使用tensorflow模型预测时,报错:FailedPreconditionError (see above for traceback): Attempting to use uninitialized value batch_normalization_14/moving_mean[[Node: batch_normalization_14/mov...原创 2019-06-21 14:29:43 · 2822 阅读 · 4 评论 -
keras强制使用cpu
安装了gpu版本后,keras预测也使用了GPU。由于项目要求测试CPU下速度,因此,在python调用代码中,做如下限制:import osos.environ["CUDA_VISIBLE_DEVICES"] = "-1"...原创 2019-06-21 13:33:44 · 4277 阅读 · 0 评论 -
tensorflow使用指定gpu
持续监控GPU使用情况命令:$ watch -n 10 nvidia-smi一、指定使用某个显卡如果机器中有多块GPU,tensorflow会默认吃掉所有能用的显存, 如果实验室多人公用一台服务器,希望指定使用特定某块GPU。可以在文件开头加入如下代码:import osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"os.envi...转载 2019-06-14 13:51:38 · 7715 阅读 · 0 评论 -
tensorflow模型预测示例
tensorflow模型预测示例:循环读入图像,预测,并统计预测时间。import tensorflow as tfimport numpy as npimport reimport osimport cv2import timemodel_dir = '/home/xuqiong/code/effcient/'#path = '/home/xuqiong/data/test...原创 2019-06-14 09:31:40 · 980 阅读 · 0 评论 -
keras转tf并加速(3)tensorflow编译后,graph_transforms工具的使用
支持的所有变换类型:add_default_attributesbackport_concatv2backport_tensor_array_v3flatten_atrous_convfold_batch_normsfold_constantsfold_old_batch_normsfreeze_requantization_rangesfuse_pad_and_convf...原创 2019-06-12 15:00:25 · 2546 阅读 · 2 评论 -
TensorFlow 图变换:FoldBatchNorm
TensorFlow 的计算是按图(Graph)组织的,构建好的图有时需要根据需要做一些变换(例如将训练好的模型部署到生产环境时,去除无用的节点),在保证计算结果不变(或近似不变)的情况下优化计算速度或降低内存占用。Graph Transform Tool【1】是 TensorFlow 提供的一组可以修改 TensorFlow Graph 的工具,使用方便,易于扩展。使用 Graph T...转载 2019-05-10 16:28:16 · 1753 阅读 · 1 评论 -
linux下安装2个版本的anaconda2和3,并安装与之对应的tensorflow
https://blog.csdn.net/wyz6666/article/details/78441596https://blog.csdn.net/qq_21578125/article/details/80245760因为需要,有时候用到Python2有时候用到Python3 ,本人之前装的是anaconda3,由于合作伙伴用的是Python2 ,为此为了考虑兼容性,经过三个小...转载 2019-05-07 21:00:23 · 1935 阅读 · 0 评论 -
keras转tf并加速(1)Keras转TensorFlow,并调用转换后模型进行预测
由于方便快捷,所以先使用Keras来搭建网络并进行训练,得到比较好的模型后,这时候就该考虑做成服务使用的问题了,TensorFlow的serving就很合适,所以需要把Keras保存的模型转为TensorFlow格式来使用。Keras模型转TensorFlow其实由于TensorFlow本身以及把Keras作为其高层简化API,且也是建议由浅入深地来研究应用,TensorFlow本身就对K...转载 2019-05-10 11:05:23 · 5180 阅读 · 1 评论 -
打印tensorflow结构
代码:import tensorflow as tfsess=tf.Sessionwith tf.Graph().as_default(): with tf.gfile.FastGFile('./models/emotion_model.pb', 'rb') as modelfile: graph_def=tf.GraphDef() graph_...原创 2019-05-10 11:03:17 · 6721 阅读 · 5 评论 -
Tensorflow 使用时cpu编译不支持警告
参考:https://blog.csdn.net/Fourierrr_/article/details/79749899whl文件:https://github.com/lakshayg/tensorflow-build https://github.com/mind/wheels使用TensorFlow模块时,弹出错误Your CPU suppor...转载 2019-05-07 16:29:14 · 535 阅读 · 0 评论 -
tensorflow: cuda driver version is insufficient for cuda runtime version
cudadriver version is insufficient for cuda runtime version :CUDA的驱动程序版本跟CUDA的运行版本不匹配背景:anaconda安装的环境1.cuda driver version(驱动版本):NVIDIA GPU的驱动程序版本查看命令:nvidia-smi我的GPU的驱动程序版本是:384.1302....原创 2019-05-07 13:59:04 · 3256 阅读 · 0 评论 -
linux下anaconda安装tensorflow gpu和keras
参考:https://blog.csdn.net/weixin_39954229/article/details/799611721、anaconda下新建环境conda create -n nosee_xception_32、激活环境并安装激活环境:conda activatenosee_xception_3安装gpu版本:conda install tensorflow-...原创 2019-05-07 11:53:25 · 989 阅读 · 0 评论 -
keras转tf并加速(2)ubuntu16.04中tensorflow压缩模型transform_graph编译
1、安装bazel官网安装指南:https://docs.bazel.build/versions/master/install-ubuntu.html#install-with-installer-ubuntu发布版本:https://github.com/bazelbuild/bazel/releasessudo apt-get install pkg-config zip g++...原创 2019-05-09 17:26:38 · 1195 阅读 · 0 评论 -
keras和tensorflow模型同时读取要慎重
项目中,先读取了一个keras模型获取模型输入size,再加载keras转tensorflow后的pb模型进行预测。报错:Attempting to use uninitialized value batch_normalization_14/moving_mean逛论坛,有建议加上初始化:sess.run(tf.global_variables_initializer())...原创 2019-07-30 15:25:12 · 873 阅读 · 0 评论