# torch==1.10.2
# torchvision==0.11.3
tensorboard==2.11.2  # pytorch need this. do not use tensorboardX
accelerate==0.20.3  # from whl
transformers==4.27.4  # for diffusers
# diffusers  # from third-party
# datasets
mmcv-full==1.4.5
hydra-core==1.3.0
einops==0.3.2
h5py>=3.8.0  # we may need version >=3.5.0. tried with 3.8.0. install from wheel
# triton  # >2.0.0post1, 2.0.0post1 is buggy, any later version should fix
# deepspeed  # >=0.9.5+1491e14e, 0.9.5 cannot work

# from bevfusion
Pillow==8.4.0
torchpack==0.3.1
nuscenes-devkit
mpi4py==3.0.3
numpy==1.21.6
numba==0.53.0  # for numpy==1.21.6
llvmlite==0.36.0
# mmdet3d  # this project
mmdet==2.20.0  # cannot run on 8 GPUs with 2.28.2
opencv-python==4.6.0.66  # visualization requires >=4.5.5.64

# cvt
pytorch-lightning==1.6.2
imgaug==0.4.0
fvcore==0.1.5.post20210825
efficientnet-pytorch==0.7.1
wandb  # for training
