diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..b6d8f82 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,18 @@ +FROM python:3.8-slim-bullseye + +LABEL maintainer "DataMade " + +RUN apt update && apt install -y gcc g++ + +RUN mkdir /app +WORKDIR /app + +# Copy the contents of the current host directory (i.e., our app code) into +# the container. +COPY . /app + +RUN pip install -e . + +RUN pip install -r requirements-dev.txt +RUN pip install -r requirements-examples.txt + diff --git a/docker-compose.yml b/docker-compose.yml new file mode 100644 index 0000000..16ed091 --- /dev/null +++ b/docker-compose.yml @@ -0,0 +1,9 @@ +services: + app: + platform: linux/amd64 + image: entity-embed + container_name: entity-embed + build: . + ports: + - 8888:8888 + command: jupyter notebook --allow-root --ip=0.0.0.0 --NotebookApp.token='' --NotebookApp.password='' diff --git a/requirements.txt b/requirements.txt index 199a67b..d660efd 100644 --- a/requirements.txt +++ b/requirements.txt @@ -8,5 +8,6 @@ pytorch-metric-learning>=0.9.98,<1.0 regex>=2020.11.13 torch>=1.7.1,<1.9 torchtext>=0.8,<0.10 -torchvision>=0.8.2<0.10 +torchvision>=0.8.2,<0.10 tqdm>=4.53.0 +six