Dockerfile.cpu

FROM tensorflow/tensorflow

# Basic setup
RUN apt-get update && apt-get install -y -q --no-install-recommends \
        apt-transport-https \
        ca-certificates \
        curl \
        wget \
        software-properties-common

# NVM environment variables
ENV NVM_DIR /usr/local/nvm
ENV NODE_VERSION 14.16.0
# Install NVM
RUN curl -o- https://raw.githubusercontent.com/creationix/nvm/v0.31.2/install.sh | bash

# Install Node and NPM
RUN . $NVM_DIR/nvm.sh \
    && nvm install $NODE_VERSION \
    && nvm alias default $NODE_VERSION \
    && nvm use default

# Add node and npm to path so the commands are available
ENV NODE_PATH $NVM_DIR/v$NODE_VERSION/lib/node_modules
ENV PATH $NVM_DIR/versions/node/v$NODE_VERSION/bin:$PATH

### Install Lux AI ###
RUN npm i -g @lux-ai/2021-challenge@latest

### QLarAI setup ###
WORKDIR /root
COPY ./requirements_cpu.txt .
RUN pip install -r requirements_cpu.txt

ADD . .

ENTRYPOINT [ "python", "train.py"]