python3.11 -m venv venv
source ./venv/bin/activate
pip install wheel
pip install --upgrade pip

conda create -n natten-fwd python=3.11
conda activate natten-fwd

pip install -r requirements.txt

git clone https://github.com/SHI-Labs/NATTEN.git
cd NATTEN
pip install cmake==3.20.3
CUDACXX=/usr/local/cuda/bin/nvcc make CUDA_ARCH="8.9" WORKERS=2