<filename>polus-vector-to-label-plugin/Dockerfile<gh_stars>1-10

FROM pytorch/pytorch:1.7.1-cuda11.0-cudnn8-runtime

COPY VERSION /
		
ARG EXEC_DIR="/opt/executables"
ARG DATA_DIR="/data"

RUN mkdir -p ${EXEC_DIR} \
    && mkdir -p ${DATA_DIR}/inputs \
    && mkdir ${DATA_DIR}/outputs

COPY src ${EXEC_DIR}/
WORKDIR ${EXEC_DIR}

RUN pip3 install -r ${EXEC_DIR}/requirements.txt --no-cache-dir

RUN mkdir ${EXEC_DIR}/numba_cache && \
    chmod 777 ${EXEC_DIR}/numba_cache && \
    mkdir /.cellpose && \
    chmod 777 /.cellpose
ENV NUMBA_CACHE_DIR=${EXEC_DIR}/numba_cache

ENTRYPOINT ["python3", "main.py"]