Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Raspberry Pi 64 OS + Jetson Nano configure.py #2420

Open
wants to merge 4 commits into
base: master
Choose a base branch
from

Conversation

Qengineering
Copy link

Description

Modified configure.py for installation on an aarch64 linux machine like Raspberry Pi 64-bit OS or Jetson Nano.
Extra lines to get CUDA and cuDNN working for the Jetson Nano.

Fixes # none

Type of change

Checklist:

  • I've properly formatted my code according to the guidelines
    • By running Black + Flake8
    • By running pre-commit hooks
  • This PR addresses an already submitted issue for TensorFlow Addons
  • I have made corresponding changes to the documentation
  • I have added tests that prove my fix is effective or that my feature works
  • This PR contains modifications to C++ custom-ops

How Has This Been Tested?

Installed the Addons successfully on the RPi 64 OS and the Jetson Nano
*

Add configuration for Raspberry Pi 64-bit OS and Jetson Nano.
The Nano does not know TF_NEED_CUDA, nor TF_CUDA_VERSION (= JETSON_CUDA).
Simplest solution is an extra branch.
@boring-cyborg boring-cyborg bot added the build label Mar 20, 2021
@google-cla
Copy link

google-cla bot commented Mar 20, 2021

Thanks for your pull request. It looks like this may be your first contribution to a Google open source project (if not, look below for help). Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

📝 Please visit https://cla.developers.google.com/ to sign.

Once you've signed (or fixed any issues), please reply here with @googlebot I signed it! and we'll verify it.


What to do if you already signed the CLA

Individual signers
Corporate signers

ℹ️ Googlers: Go here for more info.

@google-cla google-cla bot added the cla: no label Mar 20, 2021
@Qengineering
Copy link
Author

@googlebot I signed it!

@google-cla google-cla bot added cla: yes and removed cla: no labels Mar 20, 2021
@Qengineering Qengineering mentioned this pull request Apr 4, 2021
8 tasks
@StuartIanNaylor
Copy link

sudo nano configure.py Change 2 lines to the following:-

def is_raspi_arm():
return (os.uname()[4] == "armv7l") or (os.uname()[4] == "aarch64")

if is_macos() or is_linux():

write("build --copt=-mavx") <-not known in gcc aarch64

    write("build --cxxopt=-std=c++14")
    write("build --host_cxxopt=-std=c++14")

sudo ln -s /home/pi/google-kws/venv/lib/python3.7/site-packages/tensorflow/python/_pywrap_tensorflow_internal.so /usr/lib/lib_pywrap_tensorflow_internal.so

Works for Aarch64 just do the normal install after

Copy link

@StuartIanNaylor StuartIanNaylor left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pi fixes checked working nano should also

@StuartIanNaylor
Copy link

Any chance we could get this pull request merged as it doesn't actually change existing just adds support for missing architecture?

@bhack
Copy link
Contributor

bhack commented Apr 12, 2021

Tests are failing

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants