M1 Mac에서 Image data augmentation 오류 해결

Error Message

NotFoundError: No registered 'RngReadAndSkip' OpKernel for 'GPU' devices compatible with node {
{node RngReadAndSkip}} . Registered: device='XLA_CPU_JIT' device='CPU' [Op:RngReadAndSkip]

 

Solution

# 텐서를 CPU에 할당
with tf.device('/cpu:0'):
    data_augmentation = keras.Sequential(
        [
            layers.RandomFlip("horizontal"),
            layers.RandomRotation(0.1),
            layers.RandomZoom(0.2),
        ]
    )

 

tensorflow-metal 플러그인에 RngReadAndSkip Op가 아직 추가되지 않아서 발생한 문제였다. with tf.device('/cpu:0'): 코드를 한 줄 추가함으로써 해결할 수 있다. 

 

with tf.device

특정 연산을 수행할 장치를 직접 선택할 때 사용한다. 해당 컨텍스트에서의 모든 연산은 지정된 장치에서 수행된다.

 

 

Data augmentation 에러가 잘 해결된 것을 확인할 수 있다.

 

 

Reference

https://developer.apple.com/forums/thread/695319

Image data augmentation Keras M1 M… | Apple Developer Forums

https://www.tensorflow.org/guide/gpu?hl=ko 

GPU 사용하기  |  TensorFlow Core