-
Notifications
You must be signed in to change notification settings - Fork 5
/
install_tensorflow_keras_inference.sh
119 lines (108 loc) · 4.17 KB
/
install_tensorflow_keras_inference.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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
####################################################################
#
# Machine Learning with CUDA support
#
# References
# https://github.com/rbonghi/jetson_easy
#
# Urs Utzinger, Summer 2019
####################################################################
###############################################################3
# Tensorflow
###############################################################3
#sudo -H pip2 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==1.14.0+nv19.10 --user
sudo -H pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v43 tensorflow-gpu
###############################################################3
# pyTorch 1.3.0
###############################################################3
#
# https://devtalk.nvidia.com/default/topic/1049071/jetson-nano/pytorch-for-jetson-nano/
# https://devtalk.nvidia.com/default/topic/1049071/jetson-nano/pytorch-for-jetson-nano-version-1-3-0-now-available/
wget https://nvidia.box.com/shared/static/6t52xry4x2i634h1cfqvc9oaoqfzrcnq.whl -O torch-1.3.0-cp27-cp27mu-linux_aarch64.whl
sudo -H pip2 install torch-1.3.0-cp27-cp27mu-linux_aarch64.whl
#
wget https://nvidia.box.com/shared/static/phqe92v26cbhqjohwtvxorrwnmrnfx1o.whl -O torch-1.3.0-cp36-cp36m-linux_aarch64.whl
sudo -H pip3 install numpy torch-1.3.0-cp36-cp36m-linux_aarch64.whl
###############################################################3
# pytorch vision v0.5.0
###############################################################3
#
#sudo -H pip2 install protobuf --upgrade
#sudo -H pip3 install protobuf --upgrade
#
sudo -H pip2 install setuptools --upgrade
sudo -H pip3 install setuptools --upgrade
#
sudo -H pip2 install numpy --upgrade
sudo -H pip3 install numpy --upgrade
#
sudo apt-get install libjpeg-dev zlib1g-dev
# Version for tochvision needs to match pytorch version , see pyTorch link above
git clone --branch v0.5.0 https://github.com/pytorch/vision torchvision # see below for version of torchvision to download
cd torchvision
sudo -H python3 setup.py install
sudo -H python2 setup.py install
cd ../ # attempting to load torchvision from build dir will result in import error
# Test with python shell
import torch
print(torch.__version__)
print('CUDA available: ' + str(torch.cuda.is_available()))
a = torch.cuda.FloatTensor(2).zero_()
print('Tensor a = ' + str(a))
b = torch.randn(2).cuda()
print('Tensor b = ' + str(b))
c = a + b
print('Tensor c = ' + str(c))
import torchvision
print(torchvision.__version__)
###############################################################3
# Torch to RT
# https://github.com/NVIDIA-AI-IOT/torch2trt
###############################################################3
git clone https://github.com/NVIDIA-AI-IOT/torch2trt
cd torch2trt
sudo python setup.py install
###############################################################3
# Keras
###############################################################3
sudo apt-get install graphviz
sudo apt-get install graphviz-dev
sudo apt-get install libgraphviz-dev
sudo apt-get install python-pygraphviz
sudo apt-get install python3-graphviz
sudo -H pip2 install h5py
sudo -H pip3 install h5py
sudo -H pip2 install pydot
sudo -H pip3 install pydot
#
sudo apt-get -y install python3-keras
###############################################################3
# Jetson Inference
###############################################################3
sudo apt-get install doxygen
git clone https://github.com/dusty-nv/jetson-inference
cd jetson-inference
git submodule update --init
mkdir build
cd build
cmake ../
make -j4
sudo make install
# Open any of the modules that have camera version and
# edit *.cpp and set camera to 0 at #define DEFAULT_CAMERA 0 //
# check that you have a camera at /dev/video0 for camera 0
###############################################################3
#
# Face Recognition Library
#
###############################################################3
wget http://dlib.net/files/dlib-19.19.tar.bz2
tar jxvf dlib-19.19.tar.bz2
cd dlib-19.19
gedit dlib/cuda/cudnn_dlibapi.cpp
//forward_algo = forward_best_algo;
sudo python3 setup.py install
sudo -H pip3 install face_recognition
#
wget -O doorcam.py tiny.cc/doorcam
python3 doorcam.py