diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index 293c8c0..53178d2 100644
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@@ -14,7 +14,7 @@ RUN apt-get update &&  apt-get install -y  \
     pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran \
     libopenblas-dev libatlas-base-dev
 
-RUN mkdir /training && pip3 install precise-lite && pip3 install tensorflow==2.4.4 && pip3 install tensorflow-gpu==2.4.4
+RUN mkdir /training && pip3 install git+https://github.com/tystuyfzand/precise-lite.git@feature/tf24 && pip3 install tensorflow-gpu==2.4.4
 
 WORKDIR /training