


# =========== if there are no .mujoco directory ===========
# mkdir -p ~/.mujoco

# # MuJoCo 2.0.0 download (for Linux)
# wget https://www.roboti.us/download/mujoco200_linux.zip

# unzip mujoco200_linux.zip -d ~/.mujoco
# mv ~/.mujoco/mujoco200_linux ~/.mujoco/mujoco200
# wget https://www.roboti.us/file/mjkey.txt -P ~/.mujoco/
# =========== if there are no .mujoco directory ===========

echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/.mujoco/mujoco200/bin' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/{nvidia folder}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/x86_64-linux-gnu' >> ~/.bashrc

source ~/.bashrc

sudo apt update
sudo apt install -y libgl1-mesa-dev libglew-dev patchelf python3-pip python3-numpy python3-scipy

pip install mujoco-py==2.0.2.13

pip install gdown typed-argument-parser sk-video d4rl
pip install git+https://github.com/Farama-Foundation/d4rl@master#egg=d4rl

# ==========
pip install qpth
pip install torchdiffeq 
pip install torchcfm
# ===== if can't install torchcfm because of pot ====
pip install POT==0.9.4 --no-cache-dir
