publish.sh 494 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
#!/bin/bash

set -x

if [ $# -ne 2 ]; then
    REPO=dmlc
    BRANCH=master
else
    REPO=$1
    BRANCH=$2
fi

13
docker run --name dgl-reg --rm --runtime=nvidia --hostname=reg-machine -dit dgllib/dgl-ci-gpu:conda /bin/bash
14
15
docker cp ./asv_data dgl-reg:/root/asv_data/
docker cp ./run.sh dgl-reg:/root/run.sh
16
docker cp ./requirement.txt dgl-reg:/root/requirement.txt
17
18
docker exec dgl-reg bash /root/run.sh $REPO $BRANCH
docker cp dgl-reg:/root/regression/dgl/asv/. ./asv_data/
19
docker stop dgl-reg
20
21