可以在MAC上跑吗?

[taozhiyuai]

可以在MAC上跑吗?

[jingyaogong]

训练框架基本上是原生PyTorch

CPU完全支持训练

但是PyTorch对MAC的GPU支持貌似只对M1芯片及以上有效，具体如何启用可以参考：

how-to-enable-gpu-support-for-tensorflow-or-pytorch-on-macos

mac中pytorch使用

核心是把cuda换成mps

MPS 是 Apple 提供的用于 macOS 和 iOS 设备的高性能计算库，专门用于 Apple 的 GPU 上的加速计算。
CUDA 是 NVIDIA 提供的并行计算平台和 API 模型，专门用于 NVIDIA GPU 上的高性能计算。

device = torch.device("mps")
model = model.to(device)

BTW: 本人没有MAC设备供测试，不好意思~