tensorflow
月下花弄影
登山,高一步则多一层境界
展开
-
python+ tensorflow : num_elements()
方法:num_elements()功能:返回一个变量的规模,也就是该变量的所有元素的数量。这个变量应该是该类型:tensorflow.python.framework.tensor_shape.TensorShape也可以看成是tensorflow中tensor数据类型的get_shape()的返回类型代码例子:import tensorflow as tfimport numpy...原创 2018-09-25 14:51:11 · 1759 阅读 · 0 评论 -
Keras: fit_generator中如何构建一个generator?
为何使用fit_generator?在深度学习中,我们数据通常会很大,即使在使用GPU的情况下,我们如果一次性将所有数据(如图像)读入CPU的内存中,内存很有可能会奔溃。这在实际的项目中很有可能会出现。如果我们使用fit_generator则可以解决这个问题:1)fit_generator的参数中有一个是连续不断的产生数据的函数,被称为generator。2)至于这个generator是怎...原创 2018-11-04 04:22:40 · 5196 阅读 · 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 · 2565 阅读 · 3 评论