Cross-Embodiment_Manipulation_Learning/install.sh at main · FedericoMorro/Cross-Embodiment_Manipulation_Learning · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#!/bin/bash

installation_type="default"

while [[ $# -gt 0 ]]; do
    case $1 in
        --gpu)
            echo "GPU flag detected."
            installation_type="gpu"
            shift
            ;;
        --tpu)
            echo "TPU flag detected."
            installation_type="tpu"
            shift
            ;;
        *)
            echo "Unknown flag: $1"
            shift
            ;;
    esac
done


conda init
conda activate ceml_env


pip install -r requirements.txt

cd octo-pytorch
pip install -r requirements.txt
pip install -e .
cd ..

cd ggcnn
pip install -r requirements.txt
pip install -e .
cd ..

cd robosuite
pip install -r requirements.txt
pip install -r requirements-extra.txt
cd ..

cd tdmpc2
pip install -r requirements.txt
pip install -e .
cd ..

cd robomimic
pip install -r requirements.txt
pip install -e .
cd ..

pip install torch==2.6.0 torchvision


# if [[ "$installation_type" == "gpu" ]]; then
#     pip install --upgrade "jax[cuda12_pip]==0.4.20" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
#     pip install --upgrade "jaxlib[cuda12_pip]==0.4.20" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
# elif [[ "$installation_type" == "tpu" ]]; then
#     pip install --upgrade "jax[tpu]==0.4.20" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
#     pip install --upgrade "jaxlib[tpu]==0.4.20" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
# fi


current_dir=$(pwd)
export PYTHONPATH="$current_dir/robosuite:$PYTHONPATH"


python robosuite/robosuite/scripts/setup_macros.py
python robomimic/robomimic/scripts/setup_macros.py


echo "export TFDS_DATA_DIR=/home/storage/data" >> ~/.bashrc