peterlee0127 / tensorflow-nvJetson

TensorFlow for NVIDIA Jetson, also include patch and script for building.
https://tfjetson.peterlee.app
205 stars 61 forks source link

Tutorial for building tensorflow for aarch64 #12

Closed sanjaybharadwaj closed 6 years ago

sanjaybharadwaj commented 6 years ago

Hi Peter,

I appreciate that you are saving the wheels of tensorflow for aarch64 architecture. I am trying to build the same but for Python3. Could you please post me the instructions to build the tensorflow (Patches needed for Basel and other packages for the building).

I currently use Nvidia DrivePX2 as my target.

Thank you and best regards, Sanjay

peterlee0127 commented 6 years ago

With Tensorflow 1.5 and later, you can build with this guide. No need any modify anymore. Required JetPack 3.2 https://www.tensorflow.org/install/install_sources

sanjaybharadwaj commented 6 years ago

Oh so aarch64 support is added in the tensorflow ??

peterlee0127 commented 6 years ago

Yes.

https://github.com/tensorflow/tensorflow/releases/tag/v1.5.0 Add support for CUDA on NVIDIA Tegra devices