tensorflow/stream_executor/cuda/cuda_driver.cc:328] 调用 cuInit 失败：CUDA_ERROR_UNKNOWN：未知错误
我正在尝试在Tensorflow中使用GPU。我的Tensorflow版本是2.4.1，我正在使用Cuda版本11.2。这是nvidia-smi的输出。+----------------------------------------------------------------------...
谷歌协作平台在使用TensorFlow 2.0批次数据集时无法加载图像文件
一些背景信息，我正在加载大约60,000张图片到colab来训练GAN。我已经将它们上传到Drive，并且目录结构包含不同类别（大约7-8个）的文件夹在root内。我按照以下方式将它们加载到colab： root = "drive/My Drive/data/images" root = p...
TensorFlow模型.fit()使用数据集生成器
我正在使用数据集 API 生成训练数据，并将其分批排序到神经网络中。 以下是我的代码的最小工作示例： import tensorflow as tf import numpy as np import random def my_generator(): while True:...
Keras的'set_session'在Tensorflow 2.0中不可用
我将尝试完成以下任务: import tensorflow as tf from keras.models import load_model, Model from keras import backend as K sess = tf.compat.v1.Session() K.set...
生成数据集迭代器完成时发生错误：已取消：操作被取消。
在运行使用TensorFlow 2.0的代码的Kubeflow管道时，每个epoch结束时都会显示以下错误： W tensorflow/core/kernels/data/generator_dataset_op.cc:103] 在终止GeneratorDataset迭代器时发生错误：...
将Tensorflow2中的图冻结为pb文件
我们通过图形冻结将许多TF1模型部署： tf.train.write_graph(self.session.graph_def, some_path) # get graph definitions with weights output_graph_def = tf.graph_util...
无法训练我的Keras模型:（数据的基数不明确）
我将使用bert-for-tf2库来处理多类分类问题。我已经创建了模型，但训练时出现以下错误： --------------------------------------------------------------------------- ValueError ...
尝试使用StreamExecutor执行BLAS操作但没有BLAS支持，导致出现错误。
我的电脑只有1个GPU。 以下是输入某人的代码后得到的结果。 [name: "/device:CPU:0" device_type: "CPU" memory_limit: 268435456 locality {} incarnation: 16894043898758027805, n...
Tensorflow 2.0多输入自定义损失函数
我正在尝试优化使用以下两个损失函数的模型def loss_1(pred, weights, logits): weighted_sparse_ce = kls.SparseCategoricalCrossentropy(from_logits=True) policy_loss...
Tensorflow 2.0模型使用tf.function非常缓慢，并且每次训练计数更改时都会重新编译。Eager运行速度快约4倍。
我有使用未编译的Keras代码构建的模型，并尝试通过自定义训练循环运行它们。默认情况下，TF 2.0 eager代码在CPU（笔记本电脑）上运行约30秒。当我使用包装tf.function调用方法创建keras模型时，它的运行速度要慢得多，似乎需要很长时间才能启动，特别是“第一次”。例如，在t...