TF 2.0：如何转换“ tf.contrib.eager.num_gpus（）”？

Question

我发现无法在TF 2.0中使用'num_gpus'函数。 以为使用“ compat.v1”应该可以。

我在helper_function中使用了'tensorflow.contrib.eager.num_gpus（）'来初始化gpu（如果存在）。

在TF 2.0中获取所需信息的预期方式是什么？

Answer 1

使用以下SLOC代替以前的tf.contrib.eager.num_gpus（）：

from tensorflow.python.eager import context
blGPU = context.num_gpus()

if blGPU > 0 :
  with tf.device("/gpu:0"):
 ...