forked from zhygallo/clusternet_segmentation
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
35 lines (28 loc) · 1.3 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
ARG FAISS_CPU_OR_GPU=cpu
ARG FAISS_VERSION=1.4.0
ENV HOME /root
RUN apt-get update && \
apt-get install -y build-essential curl software-properties-common bzip2 python3-pip && \
curl https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh > /tmp/conda.sh && \
bash /tmp/conda.sh -b -p /opt/conda && \
/opt/conda/bin/conda update -n base conda && \
/opt/conda/bin/conda install -y -c pytorch faiss-${FAISS_CPU_OR_GPU}=${FAISS_VERSION} && \
apt-get remove -y --auto-remove curl bzip2 && \
apt-get clean && \
rm -fr /tmp/conda.sh
#RUN /opt/conda/bin/conda install -c conda-forge tensorflow-gpu \
# keras \
# scikit-image \
# click
RUN /opt/conda/bin/pip install sklearn \
scikit-image \
click \
pandas \
tensorflow-gpu==1.11.0 \
Keras==2.2.2 \
Keras-Applications==1.0.4 \
Keras-Preprocessing==1.0.2
ENV LC_ALL=C.UTF-8
ENV LANG=C.UTF-8
ENV PATH="/opt/conda/bin:${PATH}"