forked from flashlight/flashlight
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDockerfile-CPU
More file actions
19 lines (15 loc) · 750 Bytes
/
Copy pathDockerfile-CPU
File metadata and controls
19 lines (15 loc) · 750 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
# ==================================================================
# module list
# ------------------------------------------------------------------
# flashlight master (git, CPU backend)
# ==================================================================
FROM flml/flashlight:cpu-base-latest
# ==================================================================
# flashlight with CPU backend
# ------------------------------------------------------------------
# Setup and build flashlight
RUN mkdir /root/flashlight
COPY . /root/flashlight
RUN export MKLROOT=/opt/intel/mkl && cd /root/flashlight && mkdir -p build && \
cd build && cmake .. -DCMAKE_BUILD_TYPE=Release -DFLASHLIGHT_BACKEND=CPU && \
make -j8 && make install