|
FROM nvidia/cudagl:9.0-devel-ubuntu16.04
|
|
|
|
ARG DEBIAN_FRONTEND=noninteractive
|
|
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
apt-utils git curl vim unzip openssh-client wget \
|
|
build-essential cmake \
|
|
libopenblas-dev \
|
|
libglib2.0-0 \
|
|
libsm6 \
|
|
libxext6 \
|
|
libxrender-dev
|
|
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends python3.5 python3.5-dev python3-pip python3-tk && \
|
|
pip3 install --no-cache-dir --upgrade pip setuptools && \
|
|
echo "alias python='python3'" >> /root/.bash_aliases && \
|
|
echo "alias pip='pip3'" >> /root/.bash_aliases
|
|
|
|
|
|
RUN pip3 --no-cache-dir install \
|
|
numpy scipy pyyaml cffi pyyaml matplotlib Cython requests opencv-python "pillow<7"
|
|
|
|
|
|
RUN pip3 install https://download.pytorch.org/whl/cu90/torch-0.4.1-cp35-cp35m-linux_x86_64.whl && \
|
|
pip3 install torchvision==0.2.2.post3
|
|
|
|
|
|
EXPOSE 6006
|
|
|
|
|
|
RUN echo "cd /root/dense_fusion" >> /root/.bashrc
|
|
|