如何在运行tensorflow 使用GPU加速，执行代码时？

Question

在运行tensorflow 使用GPU加速，执行代码时，报错了

tensorflow.python.framework.errors_impl.InternalError: Failed copying input tensor from /job:localhost/replica:0/task:0/device:CPU:0 to /job:localhost/replica:0/task:0/device:GPU:0 in order to run _EagerConst: Dst tensor is not initialized.

Answer 1

以下答案引用自GPT-3大模型,请合理使用：
原因是，GPU加速器需要在tensorflow启动时指定device=gpu，不然就会使用CPU运行。

解决方法是在tensorflow启动时指定device=gpu，或者直接将代码中的

with tf.device("/job:localhost/replica:0/task:0/device:GPU:0"):

改为

with tf.device("gpu:0"):