vendor tensorflow-2.0.0-cp37-cp37m-linux_armv7l.whl

playbooks/bootstrap-rpi.yml

@@ -49,7 +49,7 @@
     - name: "Install platform dependencies"
       become: yes
       apt: 
-        name: ['tmux']
+        name: ['tmux', 'cmake', 'virtualenv']
     - name: "Clone rpi-vision"
       git:
         repo: 'https://github.com/leigh-johnson/rpi-vision'
@@ -103,7 +103,7 @@
         content: "/usr/local/bin/{{tft_driver_install_script}}" 
         dest: /home/pi/.tft_screen_installed
     - name: Create virtual environment
-      shell: virtualenv -p $(which python3.6) .venv
+      shell: virtualenv -p $(which python3) .venv
       args:
         creates: "{{rpi_vision_dir}}/.venv"
         chdir: "{{rpi_vision_dir}}"

setup.py

@@ -3,8 +3,9 @@
 
 '''The setup script.'''
 
-import subprocess
+import os
 import platform
+import subprocess
 from setuptools import setup, find_packages, Command
 from distutils.command.build import build as _build
 
@@ -15,23 +16,19 @@
 with open('HISTORY.rst') as history_file:
     history = history_file.read()
 
-common_requirements = [
-    'tensorflow==2.0.0'
-]
-
-# tensorflow 2.0 wheel has not been released for Raspbian yet
 trainer_requirements = [
     'ansible==2.8.1',
     'tensorflow-datasets>=1.3.0',
-    'tensorflow-hub>=0.7.0'
+    'tensorflow-hub>=0.7.0',
+    'tensorflow==2.0.0'
 ]
 trainer_requirements = list(map(
     lambda x: x + ';platform_machine=="x86_64"', trainer_requirements
 ))
 
 rpi_requirements = [
     'picamera>=1.13.0',
-    'Pillow>=6.0.0'
+    'Pillow>=6.0.0',
 ]
 rpi_requirements = list(map(
     lambda x: x + ';platform_machine=="armv7l"', rpi_requirements))
@@ -135,4 +132,8 @@ def run(self):
     url='https://github.com/leigh-johnson/rpi-vision',
     version='0.1.0',
     zip_safe=False,
+    dependency_links=[
+    # location to your egg file
+    os.path.join(os.getcwd(), 'includes', 'tensorflow-2.0.0-cp37-cp37m-linux_armv7l.whl')
+    ]
 )