| To make a local build for DeepSpeed: | |
| git clone https://github.com/microsoft/DeepSpeed/ | |
| cd DeepSpeed | |
| rm -rf build | |
| TORCH_CUDA_ARCH_LIST="8.6" DS_BUILD_CPU_ADAM=1 DS_BUILD_UTILS=1 pip install . |
| To make a local build for DeepSpeed: | |
| git clone https://github.com/microsoft/DeepSpeed/ | |
| cd DeepSpeed | |
| rm -rf build | |
| TORCH_CUDA_ARCH_LIST="8.6" DS_BUILD_CPU_ADAM=1 DS_BUILD_UTILS=1 pip install . |