diff --git a/tools/ubuntu_setup.sh b/tools/ubuntu_setup.sh
index a70def87a2..27e9d6d78e 100755
--- a/tools/ubuntu_setup.sh
+++ b/tools/ubuntu_setup.sh
@@ -89,7 +89,7 @@ pipenv install --system --deploy
 pip install -r tools/requirements.txt
 
 # to make modeld work on PC with nvidia GPU
-pip install tensorflow-gpu==2.1
+pip install tensorflow-gpu==2.2
 
 # for loggerd to work on ubuntu
 # TODO: PC should log somewhere else