-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathMakefile
56 lines (46 loc) · 1.38 KB
/
Makefile
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
PROJ_NAME=iquaflow-airplane-yolov5-use-case
CONTAINER_NAME="${PROJ_NAME}-${USER}"
ifndef DS_VOLUME
DS_VOLUME=/Nas
endif
ifndef NB_PORT
NB_PORT=8800
endif
ifndef MLF_PORT
MLF_PORT=5000
endif
help:
@echo "build -- builds the docker image"
@echo "dockershell -- raises an interactive shell docker"
@echo "notebookshell -- launches a notebook server"
@echo "mlflow -- launches an mlflow server"
@echo "downloads course materials"
build:
docker build -t $(PROJ_NAME) .
chmod 775 ./download.sh
./download.sh
dockershell:
docker run --rm --name $(CONTAINER_NAME) --gpus all \
--shm-size="32g" -p 9197:9197 \
-v $(shell pwd):/iqf -v $(DS_VOLUME):/$(DS_VOLUME) \
-it $(PROJ_NAME)
notebookshell:
docker run --gpus all --privileged -itd --rm --name $(CONTAINER_NAME)-nb \
--shm-size="32g" -p ${NB_PORT}:${NB_PORT} \
-v $(shell pwd):/iqf -v $(DS_VOLUME):/$(DS_VOLUME) \
$(PROJ_NAME) \
jupyter lab \
--NotebookApp.token='IQF' \
--no-browser \
--ip=0.0.0.0 \
--allow-root \
--port=${NB_PORT}
mlflow:
docker run --privileged -itd --rm --name $(CONTAINER_NAME)-mlf \
-p ${MLF_PORT}:${MLF_PORT} \
-v $(shell pwd):/iqf -v $(DS_VOLUME):/$(DS_VOLUME) \
$(PROJ_NAME) \
cd yolov5 && mlflow ui --host 0.0.0.0:$(MLF_PORT)
course:
wget https://image-quality-framework.s3.eu-west-1.amazonaws.com/iq-tool-box/datasets/course.zip
chmod 775 ./course.zip && unzip -o ./course.zip && rm ./course.zip