TensorflowのDatasetでtake()で異常となる

Question

念願のmacOS M1を購入したので、AIを学びたく、TensorFlow環境を下記を参考に構築しました。
PythonはJupiter Lab環境としました。
https://tech-diary.net/how-to-install-tensorflow-on-m1-mac/

先ずはサンプルを構築し理解しようと下記のサイトを例にjupyter labに入力し始めました。
https://github.com/tensorflow/docs-l10n/blob/master/site/ja/tutorials/load_data/images.ipynb

<問題>
所が、下記take()行を実行しようとするとkernelが異常終了します。
import matplotlib.pyplot as plt

plt.figure(figsize=(8,8))
for n,image in enumerate(image_ds.take(4)):
plt.subplot(2,2,n+1)
plt.imshow(image)
plt.grid(False)
plt.xticks([])
plt.yticks([])
plt.xlabel(caption_image(all_image_paths[n]))
plt.show()

<調査>
他のサイトでも同じようなサンプルがあるのでそちらでも試してみましたが、上記for ... .take(4):
で必ずエラーとなります。
単体で、image_ds.take(4)とした場合はエラーになりません。

<相談>
kernelが異常終了の原因や対応などわかりましたらご教示いただけると幸いです。

Answer 1

私のM1 Macでサンプルソースを試してみましたが、最後まで動きました。
以下のような環境です。参考にしてください。

tensorboard               2.5.0    pypi_0    pypi
tensorboard-data-server   0.6.1    pypi_0    pypi
tensorboard-plugin-wit    1.8.0    pypi_0    pypi
tensorflow-deps           2.5.0         1    apple
tensorflow-estimator      2.5.0    pypi_0    pypi
tensorflow-macos          2.5.0    pypi_0    pypi
tensorflow-metal          0.1.1    pypi_0    pypi

Comments

[fukuro]

確認ありがとうございます。当然Kernel異常となるのでメモリが足りないなどリソース関係も考えましたが、メモリは16GBにまでupしており、貴方では正常に動作確認できたことから他に要因があるのかとわかりました。

追加で確認させてください。
私のM1のTensorflow環境は以下で構築しました。
https://tech-diary.net/how-to-install-tensorflow-on-m1-mac/

記事にあるようにcondaの仮想環境で構築しました。
同じでしょうか？