-
Notifications
You must be signed in to change notification settings - Fork 1
/
docker-compose.yml
93 lines (59 loc) · 1.49 KB
/
docker-compose.yml
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
version: '2.3'
services:
notebooks:
image: jupyter/tensorflow-notebook
volumes:
- ./data:/usr/src/app/data
- ./script:/usr/src/app/script
working_dir: /usr/src/app/data
ports:
- 8888:8888
command: jupyter lab --no-browser --allow-root --ip 0.0.0.0 --NotebookApp.custom_display_url=http://localhost:8888
ipython-shell:
build: .
runtime: nvidia
volumes:
- ./data:/usr/src/app/data
- ./script:/usr/src/app/script
working_dir: /usr/src/app/data
ports:
- 6006:6006
stdin_open: true
tty: true
command: ipython --TerminalInteractiveShell.editing_mode=vi
download-images:
build: .
runtime: nvidia
volumes:
- ./data:/usr/src/app/data
- ./script:/usr/src/app/script
working_dir: /usr/src/app/data
ports:
- 6006:6006
stdin_open: true
tty: true
command: bash /usr/src/app/script/download-images.sh
create-tf-records:
build: .
runtime: nvidia
volumes:
- ./data:/usr/src/app/data
- ./script:/usr/src/app/script
working_dir: /usr/src/app/data
ports:
- 6006:6006
stdin_open: true
tty: true
command: bash /usr/src/app/script/create-tf-records.sh
train:
build: .
runtime: nvidia
volumes:
- ./data:/usr/src/app/data
- ./script:/usr/src/app/script
working_dir: /usr/src/app/data
ports:
- 6006:6006
stdin_open: true
tty: true
command: bash /usr/src/app/script/train.sh