Skip to content

Latest commit

 

History

History
36 lines (26 loc) · 602 Bytes

README.md

File metadata and controls

36 lines (26 loc) · 602 Bytes
Raw

Setup

Create virtual environment

[Option 1] virtualenv

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

[Option 2] conda

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

Install dependencies

pip install -r requirements.txt

Install NATTEN from source

You may need to install NATTEN from source. Here's how I did it:

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