forked from illinois-impact/ece408_project
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrai_build.yml
19 lines (19 loc) · 820 Bytes
/
rai_build.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
rai:
version: 0.2
image: cwpearson/ece408_mxnet_docker:amd64-gpu-2018sp-latest
resources:
cpu:
architecture: amd64
gpu:
architecture: pascal
count: 1
network: false
commands:
build:
- /bin/bash -c "cp -rv /ece408_src/* /mxnet/src/operator/custom" # copy golden files to mxnet source tree
- /bin/bash -c "cp -rv /src/* /build" # copy the project folder to /build so everything appears in the upload
- /bin/bash -c "for src in ece408_src/*; do cp -v $src /mxnet/src/operator/custom/.; done" # copy source files to mxnet tree
- nice -n20 make -C /mxnet # build mxnet
- pip2 install --user -e /mxnet/python # install python bindings
- nvprof -f -o timeline.nvvp python m1.2.py # execute code
- nvprof -f --analysis-metrics -o analysis.nvvp python m1.2.py # execute code