diff --git a/requirements-gpu.txt b/requirements-gpu.txt index 70ef8d7..576eb6a 100644 --- a/requirements-gpu.txt +++ b/requirements-gpu.txt @@ -1,4 +1,4 @@ -tensorflow-gpu==2.0.0-beta1 +tensorflow-gpu==2.5.2 pandas gradient-statsd scikit-learn