CodeSearchNet/docker/docker-gpu.Dockerfile

76 строки
1.8 KiB
Docker

FROM tensorflow/tensorflow:1.12.0-gpu-py3
ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
RUN add-apt-repository -y ppa:git-core/ppa
RUN add-apt-repository -y ppa:jonathonf/python-3.6
RUN apt-get update --fix-missing && apt-get install -y wget bzip2 ca-certificates \
byobu \
ca-certificates \
git-core git \
htop \
libglib2.0-0 \
libjpeg-dev \
libpng-dev \
libxext6 \
libsm6 \
libxrender1 \
libcupti-dev \
openssh-server \
python3.6 \
python3.6-dev \
software-properties-common \
vim \
unzip \
&& \
apt-get clean && \
rm -rf /var/lib/apt/lists/*
RUN apt-get -y update
# Setup Python 3.6 (Need for other dependencies)
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.5 1
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.6 2
RUN apt-get install -y python3-setuptools
RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
RUN python3 get-pip.py
RUN pip install --upgrade pip
# Pin TF Version on v1.12.0
RUN pip --no-cache-dir install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.12.0-cp36-cp36m-linux_x86_64.whl
# Other python packages
RUN pip --no-cache-dir install --upgrade \
altair==3.2.0 \
annoy==1.16.0 \
docopt==0.6.2 \
dpu_utils==0.1.34 \
ipdb==0.12.2 \
jsonpath_rw_ext==1.2.2 \
jupyter==1.0.0 \
more_itertools==7.2.0 \
numpy==1.16.5 \
pandas==0.25.0 \
parso==0.5.1 \
pygments==2.4.2 \
requests==2.22.0 \
scipy==1.3.1 \
SetSimilaritySearch==0.1.7 \
toolz==0.10.0 \
tqdm==4.34.0 \
typed_ast==1.4.0 \
wandb==0.8.12 \
wget==3.2
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
# Open Ports for TensorBoard, Jupyter, and SSH
EXPOSE 6006
EXPOSE 7654
EXPOSE 22
WORKDIR /home/dev/src
COPY src/docs/THIRD_PARTY_NOTICE.md .
CMD bash