diff --git a/requirements_gpu.txt b/requirements_gpu.txt index 769ebec..58427f3 100644 --- a/requirements_gpu.txt +++ b/requirements_gpu.txt @@ -15,7 +15,7 @@ scipy==1.3.0 seaborn==0.9.0 six==1.12.0 tensorboard==1.14.0 -tensorflow-gpu==1.14.0 +tensorflow-gpu==2.9.3 tf2onnx==1.5.1 torch==1.2.0 torchvision==0.4.0