diff --git a/requirements-gpu.txt b/requirements-gpu.txt index f44fc97..7db9096 100644 --- a/requirements-gpu.txt +++ b/requirements-gpu.txt @@ -4,7 +4,7 @@ Pillow cython matplotlib scikit-image -tensorflow-gpu==1.9 +tensorflow-gpu==2.12.0 keras==2.2.0 opencv-python h5py diff --git a/requirements_train.txt b/requirements_train.txt index ffb6d14..d816a4b 100644 --- a/requirements_train.txt +++ b/requirements_train.txt @@ -4,7 +4,7 @@ Pillow cython matplotlib scikit-image -tensorflow-gpu==1.9 +tensorflow-gpu==2.12.0 keras==2.2.0 opencv-python h5py