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 . |