-
-
Notifications
You must be signed in to change notification settings - Fork 466
Expand file tree
/
Copy pathDockerfile
More file actions
51 lines (46 loc) · 1.56 KB
/
Copy pathDockerfile
File metadata and controls
51 lines (46 loc) · 1.56 KB
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
FROM shareai/tensorflow:armv7l_tf1.8
RUN mkdir -p /root/.ssh
COPY ./id_rsa /root/.ssh/id_rsa
RUN chmod 600 /root/.ssh/id_rsa
COPY ./sshconfig /root/.ssh/config
RUN ls -lh && apt-get update && apt-get install -y libopenblas-base && apt-get clean && \
mkdir -p /root/.local/lib/python2.7/site-packages/
COPY ./requirements.txt /root/requirements.txt
RUN pip install -r /root/requirements.txt
RUN apt-get -y install clang
ENV CC=clang
ENV CXX=clang++
RUN git clone https://github.com/apache/incubator-mxnet.git -b 1.2.0 --recursive
RUN apt-get install -y git unzip cmake build-essential libopenblas* #liblapack* libblas*
#RUN apt-get install -y google-perftools
RUN apt-get install -y --reinstall pkg-config cmake-data
COPY assets/Makefile /incubator-mxnet/Makefile
COPY assets/CMakeLists.txt /incubator-mxnet/CMakeLists.txt
RUN mv incubator-mxnet mxnet && \
cd mxnet && \
mkdir build && \
cd build && \
cmake -DUSE_SSE=OFF \
-DUSE_F16C=OFF \
-DUSE_CUDA=OFF \
-DUSE_OPENCV=OFF \
-DUSE_OPENMP=OFF \
-DUSE_GPERFTOOLS=OFF \
-DUSE_JEMALLOC=OFF \
-DUSE_MKL_IF_AVAILABLE=OFF \
-DUSE_SIGNAL_HANDLER=ON \
-DADD_CFLAGS=-fPIC \
-DUSE_CPP_PACKAGE=OFF \
-Dmxnet_LINKER_LIBS=-latomic \
-DBUILD_CPP_EXAMPLES=OFF \
-DCMAKE_BUILD_TYPE=Release ..
#WORKDIR /root/mxnet/build
RUN pwd && ls -alh
RUN cd mxnet/build && \
make -v -j4 && \
make install
COPY assets/setup.py /mxnet/python/
RUN cd /mxnet && \
make cython PYTHON=python && \
cd /mxnet/python && \
pip install -e .