|
#!/bin/bash |
|
DOCKER_IMAGE="mirrors.tencent.com/ronnysong_rd/fastdet:torch2.0.1-cuda11.7" |
|
|
|
if [ ! -n "$DEBUG" ]; then |
|
COMMAND_PREFIX="pip3 install -e ." |
|
else |
|
COMMAND_PREFIX="pip3 install -q -e third_party/mmengine; |
|
pip3 install -q -e third_party/mmdetection; |
|
pip3 install -q -e third_party/mmcv; |
|
pip3 install -q -e third_party/mmyolo; |
|
pip3 install -q -e ." |
|
fi |
|
|
|
sudo nvidia-docker run \ |
|
--rm \ |
|
-it \ |
|
-e NVIDIA_VISIBLE_DEVICES=all \ |
|
--env="DISPLAY" \ |
|
--env="QT_X11_NO_MITSHM=1" \ |
|
--volume="$HOME/.Xauthority:/root/.Xauthority:rw" \ |
|
--shm-size=20gb \ |
|
--network=host \ |
|
-v /apdcephfs/:/apdcephfs/ \ |
|
-v /apdcephfs_cq2/:/apdcephfs_cq2/ \ |
|
-v /apdcephfs_cq3/:/apdcephfs_cq3/ \ |
|
-v /data/:/data/ \ |
|
-w $PWD \ |
|
$DOCKER_IMAGE \ |
|
bash -c "export TRANSFORMERS_CACHE=$PWD/work_dirs/.cache/transformers; |
|
export TORCH_HOME=$PWD/work_dirs/.cache/torch; |
|
export CLIP_CACHE=$PWD/work_dirs/.cache/clip; |
|
export HF_HOME=$PWD/work_dirs/.cache/hf; |
|
export TOKENIZERS_PARALLELISM=false; |
|
$COMMAND_PREFIX |
|
$*" |
|
|
