折腾了一整天终于装好了。目前,2024年8月官方发布win的两种安装方式是pip和build from source,前提要求是:The prerequisites for in both cases are Python 3.x and Pytorch with CUDA support.但是我试完的结论就是他们绝对没有做兼容性测试,包括无法import torch导致的无法编译,build from source时:
[WARNING] cpu_adam requires the 'lscpu' command, but it does not exist!
解决方案其实很简单:严格按照官方的依赖包!其实官方放出来的这张图就说明了一切。
1. 必须是python3.11,在对应PyPI release中也可以看到只有针对python3.11的版本。
2. torch 2.4.0 (cuda版本),这里有个细节,在我的电脑上必须要用pip安装在python的一个虚拟环境中,不然无法import:[Error 126],找不到指定模块。不知道这是不是个例。
3. pip更新到最新
4.最后直接pip install deepspeed 就万事大吉。