Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,29 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: mit
|
| 3 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
---
|
| 4 |
+
|
| 5 |
+
Windows wheels of [NATTEN](https://github.com/SHI-Labs/NATTEN)
|
| 6 |
+
|
| 7 |
+
Original Doc (only install, no wheel): [Buid with MSVC](https://github.com/SHI-Labs/NATTEN/blob/main/docs/install.md#Build-with-MSVC)
|
| 8 |
+
|
| 9 |
+
Build wheel steps
|
| 10 |
+
- First clone NATTEN, and make sure to fetch all submodules
|
| 11 |
+
```
|
| 12 |
+
git clone --recursive https://github.com/SHI-Labs/NATTEN
|
| 13 |
+
cd NATTEN
|
| 14 |
+
```
|
| 15 |
+
- Download WindowsWhlBuilder.bat into `NATTEN`
|
| 16 |
+
- Switch python env and make sure the corresponding torch cuda version is installed
|
| 17 |
+
|
| 18 |
+
- To build with MSVC, please open the "Native Tools Command Prompt for Visual Studio". The exact name may depend on your version of Windows, Visual Studio, and cpu architecture (in my case it was "x64 Native Tools Command Prompt for VS".)
|
| 19 |
+
```
|
| 20 |
+
# Build with 1/4th of the number of processor threads on your system
|
| 21 |
+
WindowsWhlBuilder.bat build
|
| 22 |
+
|
| 23 |
+
# Build with 8 parallel workers
|
| 24 |
+
WindowsWhlBuilder.bat build WORKERS=8
|
| 25 |
+
|
| 26 |
+
# Build targeting SM89 (Ada Lovelace)
|
| 27 |
+
WindowsWhlBuilder.bat build CUDA_ARCH=8.9
|
| 28 |
+
```
|
| 29 |
+
- Wheel file will be placed in the `dist` directory
|