forked from soumith/convnet-benchmarks
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun.sh
executable file
·21 lines (16 loc) · 1.33 KB
/
run.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
export LD_LIBRARY_PATH=/usr/local/cuda-7.0/lib:/usr/local/cuda-7.0/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=$HOME/Downloads/cudnn-6.5-linux-x64-v2:$LD_LIBRARY_PATH
export PATH="$HOME/code/bazel/bazel-bin/src":$PATH
export CUDA_VISIBLE_DEVICES=2
rm ~/code/tensorflow/tensorflow/models/convnetbenchmarks
ln -s ~/code/convnet-benchmarks/tensorflow ~/code/tensorflow/tensorflow/models/convnetbenchmarks
cd ~/code/tensorflow
bazel build -c opt --config=cuda //tensorflow/models/convnetbenchmarks:benchmark_alexnet
bazel build -c opt --config=cuda //tensorflow/models/convnetbenchmarks:benchmark_overfeat
bazel build -c opt --config=cuda //tensorflow/models/convnetbenchmarks:benchmark_vgg
bazel build -c opt --config=cuda //tensorflow/models/convnetbenchmarks:benchmark_googlenet
bazel-bin/tensorflow/models/convnetbenchmarks/benchmark_alexnet 2>&1 | tee ~/code/convnet-benchmarks/tensorflow/output_alexnet.log
bazel-bin/tensorflow/models/convnetbenchmarks/benchmark_overfeat 2>&1 | tee ~/code/convnet-benchmarks/tensorflow/output_overfeat.log
bazel-bin/tensorflow/models/convnetbenchmarks/benchmark_vgg 2>&1 | tee ~/code/convnet-benchmarks/tensorflow/output_vgga.log
bazel-bin/tensorflow/models/convnetbenchmarks/benchmark_googlenet 2>&1 | tee ~/code/convnet-benchmarks/tensorflow/output_googlenet.log
cd ~/code/convnet-benchmarks/tensorflow