diff --git a/requirements.txt b/requirements.txt index 41c20c3..dfd8f97 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,7 +4,7 @@ h5py scikit-learn torch torchvision -tensorflow-gpu==1.15.2 +tensorflow-gpu==2.9.3 mia advertorch torchxrayvision \ No newline at end of file diff --git a/requirements_new.txt b/requirements_new.txt index 4b552d5..33a130f 100644 --- a/requirements_new.txt +++ b/requirements_new.txt @@ -56,7 +56,7 @@ seaborn==0.11.1 six==1.15.0 tensorboard==1.15.0 tensorflow-estimator==1.15.1 -tensorflow-gpu==1.15.2 +tensorflow-gpu==2.9.3 termcolor==1.1.0 threadpoolctl==2.1.0 tifffile==2021.3.17