forked from BVLC/caffe
-
Notifications
You must be signed in to change notification settings - Fork 2
/
configure.sh
executable file
·44 lines (41 loc) · 1.3 KB
/
configure.sh
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
#!/usr/bin/env bash
# If CAFFE_DIR not set, try to guess
if [[ -z $CAFFE_DIR ]]; then
export CAFFE_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
fi
case `uname -n` in
(nunet)
echo Setting up for nudot...
ln -sf $CAFFE_DIR/Makefile.config.nudot $CAFFE_DIR/Makefile.config
;;
(wu)
echo Setting up for wu...
if [[ -z $WU_LARBYS_CONFIG ]]; then
source /etc/larbys.sh
fi
ln -sf $CAFFE_DIR/Makefile.config.wu $CAFFE_DIR/Makefile.config
;;
(lee)
echo Setting up for wu...
if [[ -z $LEE_LARBYS_CONFIG ]]; then
source /etc/larbys.sh
fi
ln -sf $CAFFE_DIR/Makefile.config.lee $CAFFE_DIR/Makefile.config
;;
(TowerPower)
echo Setting up for TowerPower...
if [[ -z $POWER_LARBYS_CONFIG ]]; then
source /etc/larbys.sh
fi
ln -sf $CAFFE_DIR/Makefile.config.towerpower $CAFFE_DIR/Makefile.config
;;
(hopper.nevis.columbia.edu)
echo Setting up for hopper...
ln -sf $CAFFE_DIR/Makefile.config.hopper $CAFFE_DIR/Makefile.config
;;
(*)
echo Unknown machine... Using default
ln -sf $CAFFE_DIR/Makefile.config.example $CAFFE_DIR/Makefile.config
;;
esac
export PYTHONPATH=$CAFFE_DIR/python:$PYTHONPATH