1、首先激活虚拟环境和克隆github中的DeepSpeed文件
source activate /home/Zeroad/Envs/Lora # /home/Zeroad/Envs/Lora为虚拟环境路径
git clone https://github.com/microsoft/DeepSpeed.git
2、文件下载完毕后进入DeepSpeed文件夹
cd DeepSpeed
3、进入DeepSpeed文件夹后运行install.sh文件
bash install.sh
但是在运行过程中出现报错:
此时运行以下命令解决该错误:
conda install -c nvidia cuda-compiler
4、再次运行install.sh文件
bash install.sh
5、验证是否正确安装
pip list