通用呀，就是把device从cuda换成mps，我是这么写的:
device = torch.device('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu')


【 在 xiaomeng 的大作中提到: 】
: 壕，模型代码和nv的通用吗？
--
FROM 220.248.188.*