#工作记录
在系统上安装DeepSpeed,你可以使用预编译的wheel文件来简化安装过程。
以下是四个常用的下载来源及安装教程:
一、使用daswer123/deepspeed-windows-wheels
-
根据我们的Python版本和CUDA版本,下载对应的wheel文件并可使用以下命令安装(注意修改):
例如:
- Python 3.10.x、CUDA 11.8:
pip install deepspeed-0.13.1+cu118-cp310-cp310-win_amd64.whl
二、使用AlongWY/deepspeed_wheels
-
前往Releases · AlongWY/deepspeed_wheels,找到对应版本的wheel文件。
-
下载完成后,在命令行中使用以下命令安装:
pip install deepspeed-x.x.x+cuxx-cp3xx-cp3xx-win_amd64.whl
(将
x.x.x
、xx
等替换为实际下载的版本号、CUDA版本号和Python版本号等)
三、使用S95Sedan/Deepspeed-Windows
Release Deepspeed 14.0 .whl - (python 3.11) · S95Sedan/Deepspeed-Windows
-
访问Release Deepspeed 14.0 .whl - (python 3.11) · S95Sedan/Deepspeed-Windows,下载预编译的wheel文件。
-
安装教程:
-
该版本适用于Python 3.11、CUDA 12.1,且基于“pytorch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1”编译。
-
如果你使用的是较旧版本的text-generation-webui,请使用DeepSpeed版本13.2或更低版本。
-
下载完成后,在命令行中使用以下命令安装:
pip install deepspeed-0.14.0+py311-cuda121-xx.whl
(将
xx
替换为实际的文件名) -
四、使用piwheels
-
在piwheels - deepspeed页面中找到对应版本的文件进行下载(如
deepspeed-0.16.5-py3-none-any.whl
等)。 -
下载完成后,在命令行中使用以下命令安装:
pip install deepspeed-x.x.x-py3-none-any.whl
(将
x.x.x
替换为实际下载的版本号)
通过以上方法,我们可以快速在系统上安装DeepSpeed。
在安装过程中,确保我们的Python环境、CUDA版本等与下载的wheel文件相匹配,以避免兼容性问题。
如果在安装过程中遇到权限问题,可以尝试以管理员身份运行命令提示符或使用相关命令解决。