PSA-SSL/run_docker.sh at main · TRAILab/PSA-SSL · 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
#!/bin/bash

# Read symbolic links
WAYMO_PROCESSED=$(readlink -f ./data/waymo/waymo_processed_data_v_1_2_0)
WAYMO_PROCESSED_clustered=$(readlink -f ./data/waymo/waymo_processed_data_v_1_2_0_clustered)
WAYMO_PROCESSED_labels=$(readlink -f ./data/waymo/waymo_processed_data_v_1_2_0_labels)


SEMANTIC_KITTI=$(readlink -f ./data/semantic_kitti/dataset)
NUSCENES=$(readlink -f ./data/nuscenes/v1.0-mini)

# Setup volume linking (host link:container link)
CUR_DIR=$(pwd)
PROJ_DIR=$CUR_DIR
WAYMO_PROCESSED=$WAYMO_PROCESSED:/PSA-SSL/data/waymo/waymo_processed_data_v_1_2_0
WAYMO_PROCESSED_clustered=$WAYMO_PROCESSED_clustered:/PSA-SSL/data/waymo/waymo_processed_data_v_1_2_0_clustered
WAYMO_PROCESSED_labels=$WAYMO_PROCESSED_labels:/PSA-SSL/data/waymo/waymo_processed_data_v_1_2_0_labels
SEMANTIC_KITTI=$SEMANTIC_KITTI:/PSA-SSL/data/semantic_kitti/dataset
NUSCENES=$NUSCENES:/PSA-SSL/data/nuscenes/v1.0-mini

docker run -it --env="WANDB_API_KEY=$WANDB_API_KEY" \
        --runtime=nvidia \
        --net=host \
        --privileged=true \
        --ipc=host \
        --volume="/tmp/.X11-unix:/tmp/.X11-unix:rw" \
        --volume="$HOME/.Xauthority:/root/.Xauthority:rw" \
        --env="DISPLAY" \
        --env="QT_X11_NO_MITSHM=1" \
        --hostname="inside-DOCKER" \
        --name="PSA-SSL" \
        --volume $PROJ_DIR/data:/PSA-SSL/data \
        --volume $PROJ_DIR/output:/PSA-SSL/output \
        --volume $PROJ_DIR/tools:/PSA-SSL/tools \
        --volume $PROJ_DIR/checkpoints:/PSA-SSL/checkpoints \
        --volume $PROJ_DIR/configs:/PSA-SSL/configs \
        --volume $PROJ_DIR/criterions:/PSA-SSL/criterions \
        --volume $PROJ_DIR/datasets:/PSA-SSL/datasets \
        --volume $PROJ_DIR/models:/PSA-SSL/models \
        --volume $PROJ_DIR/utils:/PSA-SSL/utils \
        --volume $WAYMO_PROCESSED \
        --volume $WAYMO_PROCESSED_clustered \
        --volume $WAYMO_PROCESSED_labels \
        --volume $SEMANTIC_KITTI \
        --volume $NUSCENES \
        psa_ssl bash