滴水恩情 发表于 2025-2-16 13:49:40

Mac M1 安装 deepspeed

目次
背景
环境
安装步骤
一、首先安装Xcode环境
二、环境准备
三、安装 ROCm 或 CUDA(可选)
四、安装 deepspeed
五、验证安装
六、运行测试
常见问题
问题1:error: subprocess-exited-with-error

背景

        在 Mac M1 上通过 pip3 安装 deepspeed 需要一些额外的步骤,因为 M1 芯片利用的是 ARM 架构,而不是传统的 x86 架构。以下是一些详细的步骤和注意事项,帮助你在 Mac M1 上成功安装 deepspeed。

环境

系统:Mac M1  macOS 13.7.2 
软件:Python3.11、Pytorch2.3.1、deepspeed0.16.2
  
安装步骤

一、首先安装Xcode环境

打开终端:


[*]按 Cmd + Space,输入 Terminal,然后按回车。
运行安装下令:
        在终端中输入如下下令:
xcode-select --install         在弹出的对话框中点击“安装”,然后按照提示完成安装。
验证安装:
        运行以下下令验证安装路径:
xcode-select -p         实行结果如下:https://i-blog.csdnimg.cn/direct/70af3098cb3d414a8329e03a49da2a0a.png
        你应该会看到类似 /Applications/Xcode.app/Contents/Developer 的输出。安装 Xcode 下令行工具后,你应该能够顺遂安装和编译 deepspeed 及其他需要编译的 Python 包。

二、环境准备



[*]安装Python
        推荐安装环境利用Anaconda举行安装和虚拟环境创建,具体安装参考如下文档:
                Mac M1 Anaconda安装 Python        


[*]安装Pytorch
        具体安装参考如下文档:
                Mac M1 安装 Pytorch、GPU加快环境        


[*] 安装deepspeed 需要一些编译工具和依靠项
        假如是brew安装的Python以及虚拟环境推荐:利用 Homebrew 安装
brew install cmake ninja py-cpuinfo
        假如是Anaconda安装的Python以及虚拟环境推荐:利用conda安装
conda install cmake ninja py-cpuinfo 实行结果如下:
https://i-blog.csdnimg.cn/direct/acb2e21e14a84bfcb4ca25a82b6fdcdc.png

三、安装 ROCm 或 CUDA(可选)

        假如你需要利用 GPU 举行加快,可以安装 ROCm 或 CUDA。不外,Mac M1 通常利用的是 Apple 的 Metal API,而不是 CUDA。你可以跳过这一步,除非你有特定的需求。
        对于利用 Apple M1 的 GPU(Metal),deepspeed 有一些特定的配置。通常情况下,deepspeed 会自动检测并利用 Metal API,但你可能需要确保相关环境变量和配置正确。

四、安装 deepspeed

        利用 pip3 安装 deepspeed。确保你利用的是正确的 Python 环境(比方虚拟环境)。
pip install deepspeed
https://i-blog.csdnimg.cn/direct/80972b9f37ec4327b2a1a4a3e2c25cf3.png

五、验证安装

        安装完成后,你可以验证 deepspeed 是否安装成功。
pip show deepspeed
这将显示 deepspeed 模块的信息。
https://i-blog.csdnimg.cn/direct/2bfd0edf77b1418e9dd0d73fdca3f3bc.png
六、运行测试

你可以运行一个简单的测试脚原来确保 deepspeed 可以正常工作。
python -c "import deepspeed; print(deepspeed.__version__)" https://i-blog.csdnimg.cn/direct/34ee5a87e9ac4c7ebc20aed865500306.png

常见问题

问题1:error: subprocess-exited-with-error

通过传统方式在mac M1中实行下令:pip install deepspeed
,直接实行如下代码,报错如下:
https://i-blog.csdnimg.cn/direct/23bb8470b56d4554b56e62be6a376ca3.png

根据提示安装py-cpuinfo库,实行如下下令:
pip install py-cpuinfo https://i-blog.csdnimg.cn/direct/b7c70a4ae8ff478394eb419e4858751c.png
实行后再实行
pip install deepspeed


免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: Mac M1 安装 deepspeed