Installation

Linux

# check nvidia driver version >= 525.60.13
cat /proc/driver/nvidia/version

# check python version is 3.10 or higher
python --verison

pip install lacss

Windows

• Make sure nividia driver version >= 452.39, following Nvidia-FAQ
• Make sure you have a python version >= 3.10 (python --version)
• Install CUDA and cudnn following Nivida's step-by-step instructions Nvidai-cudnn
• Download jaxlib wheel from jaxlib. The file name specifies versions of cuda, cudnn and python. Make sure it matches your setup.
• Install lacss and dependencies into your virtual environment

pip install <downloaded jaxlib wheel file>
pip install jax==<jaxlib version> flax==0.7.2
pip install lacss