Debug error for CPU required optimizers

最新推荐文章于 2024-08-16 18:20:57 发布

raidtest

最新推荐文章于 2024-08-16 18:20:57 发布

阅读量213

点赞数 4

文章标签：人工智能

本文链接：https://blog.csdn.net/raidtest/article/details/138330128

版权

文章描述了在使用TensorFlow时，设置global_clipnorm=1导致的错误，由于试图将Adam优化器与GPU关联，但其他节点要求CPU设备，从而引发了设备不兼容的错误。作者探讨了问题的根源和解决方法。

摘要由CSDN通过智能技术生成

optimizer = tf.keras.optimizers.Adam(learning_rate=0.00051) #, global_clipnorm=1)

this global_clipnorm = 1 ,will cause this error :

tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot assign a device for operation ResNet18_one/conv2d_20/Conv2D/ReadVariableOp: Could not satisfy explicit device specification '' because the node {{colocation_node ResNet18_one/conv2d_20/Conv2D/ReadVariableOp}} was colocated with a group of nodes that required incompatible device '/job:localhost/replica:0/task:0/device:GPU:0'. All available devices [/job:localhost/replica:0/task:0/device:CPU:0, /job:localhost/replica:0/task:0/device:GPU:0].
Colocation Debug Info:
Colocation group had the following types and supported devices:
Root Member(assigned_device_name_index_=2 requested_device_name_='/job:localhost/replica:0/task:0/device:GPU:0' assigned_device_name_='/job:localhost/replica:0/task:0/device:GPU:0' resource_device_name_='/job:localhost/replica:0/task:0/device:GPU:0' supported_device_types_=[CPU] possible_devices_=[]
_Arg: GPU CPU
ReadVariableOp: GPU CPU
ResourceApplyAdam: CPU

Colocation members, user-requested devices, and framework assigned devices, if any:
resnet18_one_conv2d_20_conv2d_readvariableop_resource (_Arg) framework assigned device=/job:localhost/replica:0/task:0/device:GPU:0
adam_adam_update_resourceapplyadam_m (_Arg) framework assigned device=/job:localhost/replica:0/task:0/device:GPU:0
adam_adam_update_resourceapplyadam_v (_Arg) framework assigned device=/job:localhost/replica:0/task:0/device:GPU:0
ResNet18_one/conv2d_20/Conv2D/ReadVariableOp (ReadVariableOp)
Adam/Adam/update/ResourceApplyAdam (ResourceApplyAdam) /job:localhost/replica:0/task:0/device:GPU:0

[[{{node ResNet18_one/conv2d_20/Conv2D/ReadVariableOp}}]] [Op:__inference_train_function_71263]