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

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

COPY . /tensorpipe

WORKDIR /tensorpipe