diff --git a/requirements-gpu.txt b/requirements-gpu.txt index 58e3a14..c2db219 100644 --- a/requirements-gpu.txt +++ b/requirements-gpu.txt @@ -53,7 +53,7 @@ sklearn==0.0 sortedcontainers==2.1.0 tensorboard==1.15.0 tensorflow-estimator==1.15.1 -tensorflow-gpu==1.15.2 +tensorflow-gpu==2.5.3 termcolor==1.1.0 torch==1.1.0 torchvision==0.3.0