![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow
月下花弄影
登山,高一步则多一层境界
展开
-
python+ tensorflow : num_elements()
方法: num_elements() 功能:返回一个变量的规模,也就是该变量的所有元素的数量。 这个变量应该是该类型: tensorflow.python.framework.tensor_shape.TensorShape 也可以看成是tensorflow中tensor数据类型的get_shape()的返回类型 代码例子: import tensorflow as tf import numpy...原创 2018-09-25 14:51:11 · 1737 阅读 · 0 评论 -
Keras: fit_generator中如何构建一个generator?
为何使用fit_generator? 在深度学习中,我们数据通常会很大,即使在使用GPU的情况下,我们如果一次性将所有数据(如图像)读入CPU的内存中,内存很有可能会奔溃。这在实际的项目中很有可能会出现。 如果我们使用fit_generator则可以解决这个问题: 1)fit_generator的参数中有一个是连续不断的产生数据的函数,被称为generator。 2)至于这个generator是怎...原创 2018-11-04 04:22:40 · 5171 阅读 · 9 评论 -
【已解决】docker + GPU + tensorflow:Tensorflow cannot open libcuda.so.1
我想在docker中使用GPU + tensorflow + keras + 一些其他的环境。 于是下载了 docker pull ufoym/deepo的镜像。 在docker run等命令后import tensorflow出现错误,错误如题。 查过很多资料,但没能解决。 最后采取了ufoym/deepo网址的启动方法,将docker run 改成nvidia-docker run,这样才可以...原创 2018-11-26 19:17:33 · 2560 阅读 · 3 评论