blob: d1ddb798971af4e859258a3b86b52fc8792a92aa (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
#!/bin/bash
#
# Wrapper to run fast_em.py using TensorFlow configured for a GPU. CUDA
# environment variables must be set.
#
# Usage:
# ./fast_em.sh <args>
set -o nounset
set -o pipefail
set -o errexit
readonly THIS_DIR=$(dirname $0)
fast-em() {
# Never returns
LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64 \
CUDA_HOME=/usr/local/cuda-7.0 \
exec $THIS_DIR/fast_em.py "$@"
}
fast-em "$@"
|