Livebook中M1 MacBook Pro上神经网络智能单元的环境配置问题解析

问题背景

在使用Livebook v0.13.3进行神经网络任务开发时，M1 MacBook Pro用户可能会遇到一个常见的环境配置问题。当尝试创建新的神经网络智能单元并安装依赖时，系统会抛出编译错误，提示XLA_TARGET环境变量设置不正确。

错误现象

具体错误信息显示：

could not compile dependency :exla
expected XLA_TARGET to be one of "cpu", "cuda", "rocm", "tpu", "cuda118", "cuda120", but got: "mps"

这表明系统检测到了一个不支持的XLA_TARGET值"mps"，而EXLA库期望的是预定义的一组值之一。

问题根源

经过分析，这个问题通常由以下原因导致：

1. 用户可能无意中在系统环境变量中设置了XLA_TARGET="mps"
2. Livebook应用内部的环境变量设置中可能包含了这个配置
3. 某些安装脚本或配置工具可能自动添加了这个设置

解决方案

要解决这个问题，可以采取以下步骤：

1. 检查Livebook环境变量设置： 在Livebook界面中，导航至设置菜单，查看环境变量部分，确认是否有XLA_TARGET的设置

2. 验证当前环境变量： 可以在Livebook中新建一个Elixir单元，执行：

   System.get_env("XLA_TARGET")
   

   查看返回值

3. 清除错误配置： 如果发现XLA_TARGET被设置为"mps"，应该将其删除或修改为支持的值之一，如"cpu"

4. 重新编译依赖： 清除配置后，可以尝试重新编译依赖：

   Mix.install([:exla], force: true)
   

技术细节

XLA（Accelerated Linear Algebra）是Google开发的线性代数编译器，EXLA是其在Elixir中的实现。XLA_TARGET环境变量用于指定要使用的计算后端：

• cpu：使用CPU进行计算
• cuda：使用NVIDIA GPU进行计算
• rocm：使用AMD GPU进行计算
• tpu：使用Google的TPU进行计算

M1芯片的Metal Performance Shaders(MPS)目前不是EXLA官方支持的后端选项，因此会导致编译失败。

最佳实践建议

1. 对于M1 Mac用户，建议使用"cpu"作为XLA_TARGET值
2. 在共享笔记本时，注意检查环境变量设置，避免将个人配置传播给其他用户
3. 定期检查Livebook的环境变量设置，确保没有冲突的配置
4. 如果需要进行GPU加速，可以考虑使用官方支持的后端或等待未来对MPS的支持

通过正确配置环境变量，M1 Mac用户可以在Livebook中顺利使用神经网络智能单元进行机器学习开发。