From ee8b808d1736390504a813533f000c7f225bfd5d Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Wed, 29 Jan 2025 14:49:05 -0800 Subject: [PATCH 1/3] Update recommended Windows whl building versions --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index cc5eb4a3500c..ec2e9a754d30 100755 --- a/setup.py +++ b/setup.py @@ -5,8 +5,8 @@ """ DeepSpeed library -To build wheel on Windows: -1. Install pytorch, such as pytorch 1.12 + cuda 11.6. +To build wheels on Windows: +1. Install pytorch, such as pytorch 2.3 + cuda 12.1. 2. Install visual cpp build tool. 3. Include cuda toolkit. 4. Launch cmd console with Administrator privilege for creating required symlink folders. From 3d6729ec53ead274765a3f1eff9e251a719bc60a Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Wed, 29 Jan 2025 15:44:25 -0800 Subject: [PATCH 2/3] Update Windows build instructions --- README.md | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 9f06af022853..9cfa155fcdee 100755 --- a/README.md +++ b/README.md @@ -192,11 +192,11 @@ of JIT compiling) or install pre-compiled ops via PyPI please see our [advanced installation instructions](https://www.deepspeed.ai/tutorials/advanced-install/). ## Windows -Windows support is partially supported with DeepSpeed. On Windows you can build wheel with following steps, currently only inference mode is supported. -1. Install pytorch, such as pytorch 1.8 + cuda 11.1 -2. Install visual cpp build tools, such as VS2019 C++ x64/x86 build tools -3. Launch cmd console with Administrator privilege for creating required symlink folders -4. Run `python setup.py bdist_wheel` to build wheel in `dist` folder +Many DeepSpeed features are supported on Windows for both training and inference. The features that are currently not supported are async io (AIO) and GDS (which does not support Windows). +1. Install PyTorch, such as pytorch 2.3+cu121. +2. Install Visual C++ build tools, such as VS2022 C++ x64/x86 build tools. +3. Launch Cmd console with Administrator permissions for creating required symlink folders and ensure MSVC tools are added to your PATH or launch the Developer Command Prompt for Visual Studio 2022 with administrator permissions. +4. Run `build_win.bat` to build wheel in `dist` folder. # Features From cda1e9567a2ff9bbcb1f6de5e402cd0c29faf910 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Wed, 29 Jan 2025 15:45:54 -0800 Subject: [PATCH 3/3] Clarify and link to blog --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 9cfa155fcdee..03aba73cfa04 100755 --- a/README.md +++ b/README.md @@ -192,7 +192,7 @@ of JIT compiling) or install pre-compiled ops via PyPI please see our [advanced installation instructions](https://www.deepspeed.ai/tutorials/advanced-install/). ## Windows -Many DeepSpeed features are supported on Windows for both training and inference. The features that are currently not supported are async io (AIO) and GDS (which does not support Windows). +Many DeepSpeed features are supported on Windows for both training and inference. You can read more about this in the original blog post [here](https://github.com/microsoft/DeepSpeed/tree/master/blogs/windows/08-2024/README.md). Among features that are currently not supported are async io (AIO) and GDS (which does not support Windows). 1. Install PyTorch, such as pytorch 2.3+cu121. 2. Install Visual C++ build tools, such as VS2022 C++ x64/x86 build tools. 3. Launch Cmd console with Administrator permissions for creating required symlink folders and ensure MSVC tools are added to your PATH or launch the Developer Command Prompt for Visual Studio 2022 with administrator permissions.