安装步骤
一开始安装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
可能的报错
报错原因:缺少了 libcudnn.so.8
共享库文件,这是由于缺少 NVIDIA cuDNN 库的原因导致的。
解决办法:安装pytorch 1.13.1+cu116_cp38 后没有加载环境
#加载环境
source /home/bingxing2/apps/package/pytorch/1.13.1+cu116_cp38/env.sh