diff --git a/requirements.txt b/requirements.txt index 0019d0d..17c41e6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,5 +6,5 @@ numpy==1.15.4 scikit-learn==0.20.2 tensorboard==1.14.0 tensorflow==1.14.0 -tensorflow-gpu==1.14.0 +tensorflow-gpu==2.12.0 tensorflow-probability==0.7.0