forked from flashlight/flashlight
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDockerfile-CUDA
More file actions
19 lines (15 loc) · 725 Bytes
/
Dockerfile-CUDA
File metadata and controls
19 lines (15 loc) · 725 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, CUDA backend)
# ==================================================================
FROM flml/flashlight:cuda-base-latest
# ==================================================================
# flashlight with GPU backend
# ------------------------------------------------------------------
# Setup and build flashlight
RUN mkdir /root/flashlight
COPY . /root/flashlight
RUN cd /root/flashlight && mkdir -p build && \
cd build && cmake .. -DCMAKE_BUILD_TYPE=Release -DFLASHLIGHT_BACKEND=CUDA && \
make -j$(nproc) && make install