diff --git a/requirements.txt b/requirements.txt old mode 100755 new mode 100644 index 34f86af..91bf698 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -tensorflow-gpu==1.14 +tensorflow-gpu==2.9.3 keras==2.1.5 ppgm==0.0.4 sklearn diff --git a/setup.py b/setup.py old mode 100755 new mode 100644 index b51de38..e3ceb24 --- a/setup.py +++ b/setup.py @@ -22,7 +22,7 @@ 'pillow', 'simpleITK', 'opencv-python', - 'tensorflow-gpu==1.14', + 'tensorflow-gpu==2.9.3', 'keras==2.1.5', 'keras-vis', 'lucid',