MacOS
MPS support
pip install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu
如果出现
(mach-o file, but is an incompatible architecture (have 'x86_64', need 'arm64e' or 'arm64')
错误,则证明安装的 torch 版本不是 arm ,需要将 torch torchvision torchaudio 删除重装。
不需要增加 --no-binary=:all: --compile --no-cache-dir --force-reinstall
这些类似的参数
pip uninstall torch torchvision torchaudio
以此类推,如果出现其他 mach-o file 的错误,也是将其卸载重装。因为 M系列芯片和 Intel 系列芯片有过渡期,可能会出现 M系列芯片的电脑上有 X86 的 packages。
比如可能会出现 numpy、PIL(其实是 Pillow)同样因为版本问题导致一样的报错。
如果是 PIL 出现问题,则需要重装 Pillow pip uninstall Pillow
-> pip install Pillow
测试安装是否成功代码
import torc