mac M 芯片下，pytorch device type

Question

小步老师，您好， 在课程的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 = ?

谢谢！

Answer 1

小布_老师 2024-07-08 21:49:21

我这边是m1芯片，没有安装rosetta2，使用代码没问题，都是我本地跑出来的。

你试试你的，我没注意到rosetta对pytorch的影响，你看下你的软件，以及python，是arm版本，还是amd版本。或者说是m系列芯片还是intel芯片

• 提问者 慕虎0285045 #1

  m1 芯片，也没有安装rosetta2，python 3.12 是arm版本。 目前选择device type = ’mpu‘ 是有上述问题。 再继续研究谢谢！

• 提问者 慕虎0285045 #2

  回复 慕虎0285045：如果是M系列芯片，在原代码将device type 改成’mps‘ 即可，速度很快！