# install problems
# pip install -r requirements.txt
h5py: pip install h5py
robomimic: pip install robomimic
jax family: pip install jax==0.4.28 jaxlib==0.4.28 \
                flax==0.8.5 orbax-checkpoint==0.5.20 \
                jax-cuda12-pjrt==0.4.28 jax-cuda12-plugin==0.4.28
wandb: setup
    def setup_wandb(
        entity='zgj21-tsinghua-university',
        project='tdmpc',
        group=None,
        name=None,
        mode='offline',
    )
mujoco-py: pip install mujoco-py==2.1.2.14
cython: pip install cython==0.29.37
robotsuite: pip install robosuite

## when Building wheel for egl_probe (setup.py) ... error
  error: subprocess-exited-with-error
pip install cmake==3.24

## jax version
Driver Version: 535.230.02   CUDA Version: 12.2  
jax family: pip install jax==0.4.29 jaxlib==0.4.29 \
                flax==0.8.5 orbax-checkpoint==0.5.20 \
                jax-cuda12-pjrt==0.4.29 jax-cuda12-plugin==0.4.29
pip install jaxlib==0.4.29+cuda12.cudnn91 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

