ben0it8

#!/usr/local/bin/python3
import logging, sys, os
logging.basicConfig(stream=sys.stdout,
                    format='%(asctime)s : %(levelname)s : %(message)s',
                    level=logging.INFO)
import torch
import torch.nn.functional as F
from pytorch_transformers import BertTokenizer
from utils import TransformerWithClfHead
from types import SimpleNamespace

ben0it8

# Install training requirements
COPY docker-res/requirements.txt "/tmp/"
RUN pip install -r "tmp/requirements.txt"

# Install app requirements
RUN pip install --upgrade uvicorn fastapi

ben0it8

# Set up paths 
ENV RESOURCES_PATH="/resources" \ 
    DATA_PATH="/data" \ 
    LOG_PATH="/logs" 
        
RUN \
    mkdir -p $RESOURCES_PATH && \
    mkdir -p $DATA_PATH && \
    chmod -R a+rwx $DATA_PATH

ben0it8

FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04

WORKDIR /srv
RUN chmod 777 /tmp

RUN apt-get update && apt-get install -y --no-install-recommends \
    build-essential \
    python3.7 python3.7-dev python3-virtualenv