如果GPU不存在，请使用CPU

Question

self.device=torc.device('cuda'if torc.cuda.is_available(else'cpu'(

上面使用Pytorch描述了检查是否有gpu的代码，如果没有，则使用cpu。如何使用tensorflow实现这一点？

Answer 1

只要安装了所有必要的库(驱动程序、CUDA等(，Tensorflow就会自动在GPU上运行，无需更改。您可以使用以下代码来检查Tensorflow是否识别您的GPU：

import tensorflow as tf
gpus = tf.config.list_physical_devices('GPU')
for gpu in gpus:
print("Name:", gpu.name, "  Type:", gpu.device_type)

如果没有可用的GPU，这个代码将只打印出任何东西，所以你知道，Tensorflow没有找到一个GPU来运行你的模型。在这种情况下，Tensorflow将在CPU上正常运行。在GPU和CPU之间切换不需要显式代码。