草庐IT

TensorFlow2

全部标签

python - 如何在 CPU 上运行 TensorFlow

我在Ubuntu14.04上安装了GPU版本的tensorflow。我在一个GPU服务器上,tensorflow可以在该服务器上访问可用的GPU。我想在CPU上运行tensorflow。通常我可以使用envCUDA_VISIBLE_DEVICES=0在GPU编号上运行。0.如何在CPU之间进行选择?我对用withtf.device("/cpu:0"):重写我的代码不感兴趣 最佳答案 也可以设置环境变量为CUDA_VISIBLE_DEVICES=""无需修改源代码。 关于python-如何

python - Tensorflow 大步论证

我正在尝试理解tf.nn.avg_pool、tf.nn.max_pool、tf.nn.conv2d中的strides参数。documentation反复说strides:Alistofintsthathaslength>=4.Thestrideoftheslidingwindowforeachdimensionoftheinputtensor.我的问题是:4+个整数分别代表什么?为什么卷积网络必须有strides[0]=strides[3]=1?在thisexample我们看到tf.reshape(_X,shape=[-1,28,28,1])。为什么是-1?遗憾的是,文档中使用-1进行

python - TensorFlow中Variable和get_variable的区别

据我所知,Variable是制作变量的默认操作,get_variable主要用于权重共享。一方面,有些人建议在需要变量时使用get_variable而不是原始的Variable操作。另一方面,我只是在TensorFlow的官方文档和演示中看到了get_variable的任何使用。因此,我想了解一些关于如何正确使用这两种机制的经验法则。有什么“标准”原则吗? 最佳答案 我建议始终使用tf.get_variable(...)-如果您需要在任何时候共享变量,这将使重构代码变得更加容易,例如在多GPU设置中(请参阅多GPUCIFAR示例)。

python - 禁用 TensorFlow 调试信息

我所说的调试信息是指TensorFlow在我的终端中显示的有关已加载库和找到的设备等的信息,而不是Python错误。Itensorflow/stream_executor/dso_loader.cc:105]successfullyopenedCUDAlibrarylibcublas.solocallyItensorflow/stream_executor/dso_loader.cc:105]successfullyopenedCUDAlibrarylibcudnn.solocallyItensorflow/stream_executor/dso_loader.cc:105]succe

python - 如何在 tensorflow 中获取当前可用的 GPU?

我有一个使用分布式TensorFlow的计划,我看到TensorFlow可以使用GPU进行训练和测试。在集群环境中,每台机器可能有0个或1个或更多GPU,我想在尽可能多的机器上将我的TensorFlow图运行到GPU中。我发现在运行tf.Session()时,TensorFlow会在日志消息中提供有关GPU的信息,如下所示:Itensorflow/core/common_runtime/gpu/gpu_init.cc:126]DMA:0Itensorflow/core/common_runtime/gpu/gpu_init.cc:136]0:YItensorflow/core/comm

docker - 带有 TensorFlow 后端的 Keras 不使用 GPU

我构建了docker镜像的gpu版本https://github.com/floydhub/dl-docker使用keras版本2.0.0和tensorflow版本0.12.1。然后我运行了mnist教程https://github.com/fchollet/keras/blob/master/examples/mnist_cnn.py但意识到keras没有使用GPU。以下是我的输出root@b79b8a57fb1f:~/sharedfolder#pythontest.pyUsingTensorFlowbackend.Downloadingdatafromhttps://s3.amaz

python - 如何将数据读入 TensorFlow?

我正在尝试将数据从CSV文件读取到tensorflow,https://www.tensorflow.org/versions/r0.7/how_tos/reading_data/index.html#filenames-shuffling-and-epoch-limits官方文档中的示例代码是这样的:col1,col2,col3,col4,col5=tf.decode_csv(value,record_defaults=record_defaults)要读取文件,需要事先知道文件有多少列和多少行,如果有1000列,我需要定义1000个变量,如col1,col2,col3,col4,c

python - TensorFlow 用户应该更喜欢 SavedModel 而不是 Checkpoint 或 GraphDef?

来自SavedModelDocs,SavedModel,theuniversalserializationformatforTensorFlowmodels.和SavedModelwrapsaTensorFlowSaver.TheSaverisprimarilyusedtogeneratethevariablecheckpoints.据我了解,如果有人想使用TensorFlowServing,SavedModel是必须的。但是,我可以在没有SavedModel的情况下将Tensorflow模型部署到服务服务器:卡住图形并将其导出为GraphDef,然后使用ReadBinaryProto

解决from tensorflow.contrib import layers报错,No module named ‘tensorflow.contrib‘的问题

这个问题出现的时候,首先要检查自己安装的tensorflow包的版本,因为一般tensorflow大于2.0的版本已经不再支持contrib的应用。首先打开电脑命令行模式,输入pythonimporttensorflowastftf.__version__查看电脑tensorflow版本,如果版本大于2.0,这时候就衍生出两种解决方法 1:手动降低tensorflow的版本。首先在在AnacondaPrompt中输入:pipuninstalltensorflow卸载原有版本 tensorflow。然后输入pipinstalltensorflow==1.13.1解决问题。2:新版本tensorf

50.现有移动端开源框架及其特点—FeatherCNN与TensorFlow Lite

50.1FeatherCNNFeatherCNN是由腾讯AI平台部研发的基于ARM架构的高效CNN推理库,该项目支持Caffe模型,且具有高性能、易部署、轻量级三大特性。该项目具体特性如下:高性能:无论是在移动设备(iOS/Android),嵌入式设备(Linux)还是基于ARM的服务器(Linux)上,FeatherCNN均能发挥最先进的推理计算性能;易部署:FeatherCNN的所有内容都包含在一个代码库中,以消除第三方依赖关系。因此,它便于在移动平台上部署。FeatherCNN自身的模型格式与Caffe模型完全兼容。轻量级:编译后的FeatherCNN库的体积仅为数百KB。50.2Ten