diff --git a/requirements-gpu.txt b/requirements-gpu.txt index 86a7cd3..68375a3 100644 --- a/requirements-gpu.txt +++ b/requirements-gpu.txt @@ -29,6 +29,6 @@ scipy==1.1.0 six==1.11.0 subprocess32==3.5.2 tensorboard==1.10.0 -tensorflow-gpu==1.10.1 +tensorflow-gpu==2.7.2 termcolor==1.1.0 Werkzeug==0.14.1