Hannes Kuchelmeister
commited on
Commit
·
554c212
1
Parent(s):
3e2d18e
add dockerfile for model
Browse files- model/.dockerignore +4 -0
- model/Dockerfile +13 -0
- model/docker-compose.yml +12 -0
- model/requirements.txt +2 -1
model/.dockerignore
ADDED
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
1 |
+
.env
|
2 |
+
in/
|
3 |
+
out/
|
4 |
+
new_proect.py
|
model/Dockerfile
ADDED
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
FROM python:3.7
|
2 |
+
|
3 |
+
WORKDIR /usr/src/app
|
4 |
+
|
5 |
+
RUN apt-get update
|
6 |
+
RUN apt-get install libgl1 -y
|
7 |
+
|
8 |
+
COPY requirements.txt ./
|
9 |
+
RUN pip install --no-cache-dir -r requirements.txt
|
10 |
+
|
11 |
+
COPY . .
|
12 |
+
|
13 |
+
CMD sh -c "python train.py --config config.json"
|
model/docker-compose.yml
ADDED
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
version: "3" # optional since v1.27.0
|
2 |
+
services:
|
3 |
+
model:
|
4 |
+
build: .
|
5 |
+
ports:
|
6 |
+
- 6006:6006
|
7 |
+
volumes:
|
8 |
+
- ./saved:/usr/src/app/saved:z
|
9 |
+
# - ./out/:/usr/src/app/out:z
|
10 |
+
# - ./in/:/usr/src/app/in:z
|
11 |
+
#env_file:
|
12 |
+
# - .env
|
model/requirements.txt
CHANGED
@@ -2,4 +2,5 @@ torch>=1.1
|
|
2 |
torchvision
|
3 |
numpy
|
4 |
tqdm
|
5 |
-
|
|
|
|
2 |
torchvision
|
3 |
numpy
|
4 |
tqdm
|
5 |
+
tensorboardx
|
6 |
+
pandas
|