TensorFlow2.0
baidu_huihui
这个作者很懒,什么都没留下…
展开
-
Ubuntu18.04 + CUDA10.0 + tensorflow-gpu 安装过程
Ubuntu18.04 + CUDA10.0 + tensorflow-gpu 安装过程简介随着软件不断更新,现在Ubuntu18.04逐渐成为客户端的主流,加上tf开始支持CUDA10,在这里在更新一下教程。看一下tensorflow官方版本的支持:确定安装1.13.1的版本安装CUDA10.0下载CUDA10:https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Linux&target_arch原创 2020-09-09 18:07:48 · 2700 阅读 · 0 评论 -
TFRecord格式介绍,TFRecord格式是一种二进制文件
TFRecord格式TFRecord文件中的数据都是通过tf.train.Example Protocol Buffer的格式存储的,TFRecord格式是一种二进制文件,它能够更好的利用内存,更方便复制和移动,并且不需要单独的标签文件;我们可以写一段代码获取你的数据,然后将数据填入到Example协议内存块(protocol buffer),将协议内存块序列化为一个字符串,并且通过tf.python_io.TFRecordWriter写入到TFRecord文件中去从TFRecords文件中读...原创 2020-08-10 17:16:16 · 763 阅读 · 0 评论 -
TensorFlow2.0 CUDA_ERROR_OUT_OF_MEMORY out of memory 解决办法
今天头一次使用TensorFlow2.0的GPU版本进行训练,训练中遇到【CUDA_ERROR_OUT_OF_MEMORY out of memory】,于是百度各种解决办法,基本能查到的都是TensorFlow1.0的用法,但此方法在TensorFlow2.0中不适用。于是经过多方查找,终于找到设置方法,在此进行记录,一是方便日后回看,二是弥补百度中各种查不到。TensorFlow1.0用法 from keras import backend as Kconfig = t...原创 2020-08-10 16:48:07 · 2903 阅读 · 5 评论 -
tensorflow2.0 GPU显存设置
遇到一个问题:新买显卡显存8G但是tensorflow运行的时候界面显示只有约6.3G的显存可用,如下图:即限制了我的显存,具体原因为什么我也不知道,但原来的显存小一些的显卡就没有这个问题。目前的解决办法是:官方文档解决然后对应的中文博客总结一下,就是下面的两个办法: 设置每次使用时按需申请 # 获取所有GPU组成listgpus = tf.config.experimental.list_physical_devices('GPU')# 设置按需申请# 由于我这里仅有一...转载 2020-08-08 02:56:57 · 2769 阅读 · 2 评论