详解TensorFlow报”UnimplementedError: RandomCrop is only supported on CPU “的原因以及解决办法

问题描述

当你试图在 TensorFlow 模型中使用 RandomCrop 函数时，你可能会遇到以下错误：

UnimplementedError: RandomCrop is only supported on CPU.

其实，这个错误是由于 RandomCrop 函数只支持 CPU 的缘故。

解决办法

如果你的模型需要使用 RandomCrop 函数，那么你需要将模型的操作放在 CPU 上执行。你可以通过设置环境变量来实现这个目的，具体操作如下：

在模型开始运行之前，先设置环境变量：

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

其中，TF_CPP_MIN_LOG_LEVEL 的值为 "2"，表示只输出错误信息；CUDA_VISIBLE_DEVICES 的值为 "-1"，表示不使用 GPU。

将模型的操作设置在 CPU 上执行：

tf.config.threading.set_inter_op_parallelism_threads(1)
tf.config.threading.set_intra_op_parallelism_threads(1)

其中，inter_op_parallelism_threads 表示 TensorFlow 执行操作的线程数，即并行度；intra_op_parallelism_threads 表示 TensorFlow 执行单个操作时使用的线程数。在这里，我们将它们的值都设置为 1，表示所有操作都将在一个线程上执行，即运行在 CPU 上。

最后，重新运行代码，这个问题就可以被解决了。