Centos7环境下成功安装DeepSpeed包

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