M1(ARM64) Tensorflow 설치중 에러 Node: 'StatefulPartitionedCall_212' could not find registered platform with id: 0x10879c220 [[{{node StatefulPartitionedCall_212}}]] [Op:__inference_train_function_23355]

Seungsoo Lee·2023년 2월 18일
0

data science

목록 보기
3/4

https://developer.apple.com/metal/tensorflow-plugin/
이 과정을 따라하다가 4. Verify 에서 문제가 생겼다 알고보니

  1. Install base TensorFlow
    python -m pip install tensorflow-macos==2.9
  2. Install tensorflow-metal plug-in
    python -m pip install tensorflow-metal==0.5.0

버전 이슈였다.

그리고 python 3.9로 하는것을 추천한다

Node: 'StatefulPartitionedCall_212' could not find registered platform with id: 0x10879c220 [[{{node StatefulPartitionedCall_212}}]][Op:__inference_train_function_23355]

이 에러가 뜬다.

1개의 댓글

comment-user-thumbnail
2023년 3월 4일

와 감사합니다! 덕분에 해결되었어요~

Metal device set to: Apple M1 Max

systemMemory: 64.00 GB
maxCacheSize: 24.00 GB

2023-03-04 21:48:19.045153: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:305] Could not identify NUMA node of platform GPU ID 0, defaulting to 0. Your kernel may not have been built with NUMA support.
2023-03-04 21:48:19.045297: I tensorflow/core/common_runtime/pluggable_device/pluggable_device_factory.cc:271] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 0 MB memory) -> physical PluggableDevice (device: 0, name: METAL, pci bus id: <undefined>)
2023-03-04 21:48:21.101263: W tensorflow/core/platform/profile_utils/cpu_utils.cc:128] Failed to get CPU frequency: 0 Hz
Epoch 1/5
/Users/a202101054/PycharmProjects/tf_learning2/venv/lib/python3.10/site-packages/tensorflow/python/util/dispatch.py:1082: UserWarning: "`sparse_categorical_crossentropy` received `from_logits=True`, but the `output` argument was produced by a sigmoid or softmax activation and thus does not represent logits. Was this intended?"
  return dispatch_target(*args, **kwargs)
2023-03-04 21:48:23.278108: I tensorflow/core/grappler/optimizers/custom_graph_optimizer_registry.cc:113] Plugin optimizer for device_type GPU is enabled.
782/782 [==============================] - 61s 68ms/step - loss: 5.0685 - accuracy: 0.0498
Epoch 2/5
782/782 [==============================] - 51s 65ms/step - loss: 4.4733 - accuracy: 0.0766
Epoch 3/5
782/782 [==============================] - 52s 67ms/step - loss: 4.3740 - accuracy: 0.0817
Epoch 4/5
782/782 [==============================] - 52s 67ms/step - loss: 4.1801 - accuracy: 0.0901
Epoch 5/5
782/782 [==============================] - 49s 63ms/step - loss: 3.8292 - accuracy: 0.1325
답글 달기