在Jupyter中导入TensorFlow时内核重启问题求助

问题概述 #

• Jupyter Notebook运行以下代码时内核崩溃，提示：The kernel appears to have died. It will restart automatically.

  import tensorflow as tf
  import numpy as np
  from tensorflow import keras
  

• 终端运行同名.py文件时报错：zsh: illegal hardware instruction python fileName.py
• 环境：Mac M1（32GB内存），已尝试重装所有包及Anaconda Navigator，问题未解决

解决方案 #

M1/M2系列Mac为ARM架构，普通x86版本的TensorFlow会因架构不兼容引发上述问题，需安装Apple官方适配版本：

1. 创建并激活专属conda环境 #

conda create -n tf_m1 python=3.9
conda activate tf_m1

（Python版本推荐3.8-3.10，此区间版本适配性最佳）

2. 安装Apple适配的TensorFlow依赖 #

conda install -c apple tensorflow-deps

3. 安装TensorFlow核心包及GPU加速插件 #

python -m pip install tensorflow-macos
python -m pip install tensorflow-metal

tensorflow-metal用于调用M1的GPU算力，同时解决架构不兼容导致的崩溃问题。

4. 验证安装有效性 #

在激活的环境中运行以下代码：

import tensorflow as tf
print(tf.__version__)
print(tf.config.list_physical_devices())

若正常输出版本号及包含GPU的设备列表，说明安装成功。

5. 配置Jupyter Notebook使用该环境 #

python -m pip install ipykernel
python -m ipykernel install --user --name=tf_m1

打开Jupyter Notebook后，通过Kernel > Change Kernel选择tf_m1环境即可正常运行代码。

注意事项 #

• 禁止使用Anaconda默认源的TensorFlow包，此类包为x86架构编译，无法在M1上正常运行。
• 全程需保证在tf_m1环境中操作，避免与其他环境的依赖冲突。

内容的提问来源于stack exchange，提问作者Gothiquo