安装步骤

一开始安装deepspeed不可以使用pip直接进行安装。

这时我们需要利用git进行clone下载到本地:

git clone https://github.com/microsoft/DeepSpeed.git

进入到deepspeed的安装目录下

cd /home/bingxing2/ailab/group/ai4agr/wzf/Tools/DeepSpeed

激活你的环境

source activate myenv

安装deepspeed

bash install.sh

 可能的报错

 1.

报错原因:缺少了 libcudnn.so.8 共享库文件,这是由于缺少 NVIDIA cuDNN 库的原因导致的。

解决办法:安装pytorch 1.13.1+cu116_cp38 后没有加载环境

#加载环境
source /home/bingxing2/apps/package/pytorch/1.13.1+cu116_cp38/env.sh

 

参考:windows安装python库 DeepSpeed常见问题_deepspeed安装-CSDN博客

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐