lldacing/NATTEN-windows · Hugging Face

Windows wheels of NATTEN

Original Doc (only install, no wheel): Buid with MSVC

Build wheel steps

First clone NATTEN, and make sure to fetch all submodules

git clone --recursive https://github.com/SHI-Labs/NATTEN
cd NATTEN

Switch tag branch, such as v0.17.3 (you can get latest tag by git describe --tags or list all available tags by git tag -l)

git checkout -b v0.17.3 v0.17.3

Download WindowsWhlBuilder.bat into NATTEN
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 2022".)

My Visual Studio Installer version
Switch python env and make sure the corresponding torch cuda version is installed
Start task

# Build with 1/4th of the number of processor threads on your system
WindowsWhlBuilder.bat build 

# Build with 8 parallel workers
WindowsWhlBuilder.bat build WORKERS=8

# Build targeting SM80 (I use this. You can also add WORKERS paramter)
WindowsWhlBuilder.bat build CUDA_ARCH=8.0

Wheel file will be placed in the dist directory

Test

Install natten

cd dist
# replace your real wheel name
pip install natten_wheel_name

Return NATTEN work path and run test case
```
cd ..
WindowsWhlBuilder.bat test
```