This implementation requires Python 3.11 and PyTorch 2.0.1. Install the dependencies We recommend using cuda 11.8(at least). conda create -n $YOUR_PY_ENV_NAME python ...