用于 Ubuntu VM 的 tensorflow-metal -- Apple M2

Question

我有一台配备 M2 芯片的 MacBook Pro。我正在使用 TensorFlow 开发、训练和测试模型。我想利用 GPU 的 tensorflow-metal 来加速训练。然而，由于其他软件要求，模型的训练必须在 Ubuntu 20.04 中完成，因此我使用 Parallels 启动了一个 VM。

如何安装 tensorflow-metal（或类似的东西）以便我可以在训练时使用我的 GPU（在 Ubuntu VM 上）？

版本信息：

$ python3
Python 3.8.10 (default, Mar 13 2023, 10:26:41) 
[GCC 9.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> print(tf.__version__)
2.12.0

需要我使用Ubuntu 20.04的软件：

• ROS Noetic
• 凉亭 11
• PX4

我也愿意使用 PyTorch 或尝试做一些事情，比如在多个操作系统上运行 ROS 节点（还不确定该怎么做）。