TensorFlow全局设置可见GPU编号与显存管理指南

本文详解在TensorFlow中全局限制可见GPU编号的两种核心方法:通过ConfigProto配置visible_device_list以及直接设置环境变量CUDA_VISIBLE_DEVICES。针对TensorFlow 2.x,进一步介绍了利用tf.config.experimental API实现指定GPU可见、逻辑分区(VirtualDeviceConfiguration)及显存动态增长(memory_growth)的高级配置技巧,帮助开发者在多GPU环境中优化资源分配与模型部署效率。

解决Windows下TensorFlow CPU转GPU安装报错及环境配置指南

本文详细记录了在Windows系统中将TensorFlow从CPU版本迁移至GPU版本时遇到的权限拒绝及DLL加载失败问题。重点分析了CUDA版本不兼容导致的错误,并提供了完整的解决步骤,包括清理残留文件、调整Visual Studio版本及正确配置CUDA 10.0与cuDNN环境,最终实现TensorFlow 1.13在Python 3.7下的成功运行。

解决Windows下TensorFlow CPU转GPU安装报错及环境配置指南

TensorFlow中如何确定张量的形状实例

本文介绍TensorFlow中获取张量形状的方法。tf.shape()可在计算图中动态获取形状,tf.shape_n()可获取多个张量形状,tf.size()获取元素个数,tf.rank()确定秩。注意tensor.get_shape()和tensor.shape仅在构建图时有效,无法在sess.run中执行。

TensorFlow 2.0模型转为TF 1.x PB格式的方法

本文介绍在TensorFlow 2.0环境下训练模型后,如何将其转换为适用于TF 1.x的.pb模型文件。针对TF 2.0不再保存.meta文件的情况,提供了一种通过Keras中间层进行转换的方法。该方法要求模型结构完全基于tf.keras构建,且仅保存权重文件(.h5)。通过定义模型结构并加载权重,在TF 1.x环境中冻结计算图,生成可直接用于推理或进一步转换的.pb文件。

TensorFlow 2.0模型转为TF 1.x PB格式的方法