-
Notifications
You must be signed in to change notification settings - Fork 4
/
build
executable file
·30 lines (25 loc) · 972 Bytes
/
build
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
#! /bin/bash
CWD = "$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
mkdir JetsonGPIO/build && cd JetsonGPIO/build
cmake .. -DCMAKE_INSTALL_PREFIX=${CWD}/JetsonGPIO/install
make install
cd libi2c && make libi2c.a
mkdir -p libi2c/install/include && mkdir libi2c/install/lib
cp libi2c/libi2c.a libi2c/install/lib
cp -r libi2c/include/i2c libi2c/install/include/
./onnxruntime/build.sh \
--use_cuda \
--cuda_home /usr/local/cuda \
--cudnn_home /usr/lib/aarch64-linux-gnu \
--use_tensorrt \
--tensorrt_home /usr/lib/aarch64-linux-gnu \
--config RelWithDebInfo \
--build_shared_lib \
--build \
--build_wheel \
--update \
--skip_submodule_sync \
--skip_tests \
--arm64 \
--cmake_extra_defines ONNXRUNTIME_VERSION=1.9.1 CMAKE_INSTALL_PREFIX=~/code/ZeroShotRT/onnxruntime/install CUDA_CMAKE_COMPILE=/usr/local/cuda/nvcc
docker build -t zeroshotrt .