setup.sh

#!bin/bash
# on a fresh pi:
pip3 install virtualenv
cd ~
virtualenv tf2_stable_baselines
cd tf2_stable_baselines/bin
source ./activate
cd ../../RealDilemma
pip3 install -r requirements.txt
pip3 install -U numpy

# install local packages
cd Pi
pip install -e  PiBot # PiBot is also a package in pip so need to be careful bout this
cd ../src/DRL
pip install -e gym-pibot

# https://github.com/sophiagu/stable-baselines-tf2
# find the installation of stable_baselines
cd ~
path=$(find . -type d -name "stable_baselines")
echo $path
rm -r $path
# install tf2 version
git clone https://github.com/sophiagu/stable-baselines-tf2
# copy the installation and remove from download location
mkdir $path
cp -r stable-baselines-tf2/* $path
sudo rm -r stable-baselines-tf2

# lastly install tf2
pip3 install https://github.com/bitsy-ai/tensorflow-arm-bin/releases/download/v2.4.0-rc2/tensorflow-2.4.0rc2-cp37-none-linux_armv7l.whl