FROM nvidia/cuda:10.2-devel-ubuntu18.04

# Install APT packages.
RUN apt-get update && \
        apt-get install -y build-essential cmake

COPY . /tensorpipe

WORKDIR /tensorpipe