File tree 1 file changed +5
-1
lines changed
1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -47,13 +47,17 @@ TRITON_DIR=${TRITON_DIR:="/opt/tritonserver"}
47
47
ARCH=${ARCH:= " x86_64" }
48
48
SERVER=${TRITON_DIR} /bin/tritonserver
49
49
BACKEND_DIR=${TRITON_DIR} /backends
50
- DATADIR=${DATADIR:= " /data/inferenceserver/${REPO_VERSION} " }
51
50
MODEL_REPO=" ${PWD} /models"
52
51
PERF_CLIENT=../clients/perf_client
53
52
TF_VERSION=${TF_VERSION:= 2}
54
53
SERVER_ARGS=" --model-repository=${MODEL_REPO} --backend-directory=${BACKEND_DIR} --backend-config=tensorflow,version=${TF_VERSION} "
55
54
source ../common/util.sh
56
55
56
+ # DATADIR is already set in environment variable for aarch64
57
+ if [ " $ARCH " != " aarch64" ]; then
58
+ DATADIR=" /data/inferenceserver/${REPO_VERSION} "
59
+ fi
60
+
57
61
# Select the single GPU that will be available to the inference server
58
62
export CUDA_VISIBLE_DEVICES=0
59
63
You can’t perform that action at this time.
0 commit comments