rei-forks/precise-lite-amd64aarch64
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Support installing tensorflow on other Python versions when on a pi

Browse Source
This commit is contained in:
Matthew D. Scholefield 2018-04-20 12:03:43 -05:00
parent b74e80568b
commit ffcf4a0e9d
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
setup.sh
View File

@ -24,6 +24,19 @@ wait_for_apt() {
vpython() { "$VENV/bin/python" $@; }
vpip() { "$VENV/bin/pip" $@; }
install_tensorflow_armv7l() {
maj_min=$(python3 -c "import sys; i = sys.version_info; print(str(i[0]) + str(i[1]))")
whl=tensorflow-1.1.0-cp34-cp34m-linux_armv7l.whl
ver_whl=${whl//34/$maj_min}
url=https://github.com/samjabrahams/tensorflow-on-raspberry-pi/releases/download/v1.1.0/$whl
wget "$url" -O $ver_whl
vpip install "$ver_whl"
vpip uninstall mock || true; vpip install mock
rm "$ver_whl"
}
#############################################
set -e; cd "$(dirname "$0")" # Script Start #
#############################################
@ -42,12 +55,8 @@ fi
arch="$(python3 -c 'import platform; print(platform.machine())')"
if ! vpython -c 'import tensorflow' 2>/dev/null && [ "$arch" = "armv7l" ]; then
whl=tensorflow-1.1.0-cp34-cp34m-linux_armv7l.whl
wget "https://github.com/samjabrahams/tensorflow-on-raspberry-pi/releases/download/v1.1.0/$whl"
vpip install "$whl"
vpip uninstall mock || true; vpip install mock
rm "$whl"
if [ "$arch" = "armv7l" ] && ! vpython -c 'import tensorflow' 2>/dev/null; then
install_tensorflow_armv7l
fi
vpip install -e runner/