mac M 芯片下，pytorch device type
来源：20-7 亲自动手训练一个AI模型（2）
慕虎0285045
2024-07-08
小步老师，您好， 在课程的20-6与20-7的Pytorch代码讲解很有收获。但其中您提到（我查阅文档Pytorch 1.1 之后就支持）可以使用device type = ‘mpu’,但是碰到以下runtime error
device = torch.device(‘mpu’) # cuda mpu
^^^^^^^^^^^^^^^^^^^
RuntimeError: Expected one of cpu, cuda, ipu, xpu, mkldnn, opengl, opencl, ideep, hip, ve, fpga, ort, xla, lazy, vulkan, mps, meta, hpu, mtia, privateuseone device type at start of device string: mpu
我也查询了Gemma 2 ， 提到使用device type ‘cuda:0’, 但是trains_module.py 还是出现其他代码错误。
因此，想跟老师在确认，mac M 芯片下，pytorch device type = ?
谢谢！
写回答
1回答
-
小布_老师
2024-07-08
我这边是m1芯片，没有安装rosetta2，使用代码没问题，都是我本地跑出来的。
你试试你的，我没注意到rosetta对pytorch的影响，你看下你的软件，以及python，是arm版本，还是amd版本。或者说是m系列芯片还是intel芯片
022024-07-09
相似问题