I try to solve the environment as you mentioned in the readme. But i have some bug.
/root/miniconda3/envs/flowformer/lib/python3.7/site-packages/torch/cuda/init.py:125: UserWarning:
NVIDIA L20 with CUDA capability sm_89 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the NVIDIA L20 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/