diff --git a/requirements.txt b/requirements.txt index 023c032..11de3c9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,6 +4,6 @@ pillow==9.0.0 scipy==1.1.0 sklearn==0.0 tensorflow==1.12.0 -tensorflow-gpu==1.14.0 +tensorflow-gpu==2.7.2 torch==1.0.1 torchvision==0.2.1