diff --git a/requirements-colab.txt b/requirements-colab.txt index 7c69191..b064f82 100644 --- a/requirements-colab.txt +++ b/requirements-colab.txt @@ -2,7 +2,7 @@ numpy==1.16.3 h5py==2.9.0 Keras==2.2.4 opencv-python==4.0.0.21 -tensorflow-gpu==1.13.1 +tensorflow-gpu==2.7.2 plaidml-keras==0.5.0 scikit-image tqdm diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 5e50c09..5f4aaa3 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -2,7 +2,7 @@ numpy==1.16.3 h5py==2.9.0 Keras==2.2.4 opencv-python==4.0.0.21 -tensorflow-gpu==1.12.0 +tensorflow-gpu==2.7.2 plaidml==0.6.0 plaidml-keras==0.5.0 scikit-image diff --git a/requirements_colab.txt b/requirements_colab.txt index fc905e1..91ad608 100644 --- a/requirements_colab.txt +++ b/requirements_colab.txt @@ -2,7 +2,7 @@ numpy==1.16.1 h5py==2.9.0 Keras==2.2.4 opencv-python==4.0.0.21 -tensorflow-gpu==1.13.1 +tensorflow-gpu==2.7.2 plaidml-keras==0.5.0 scikit-image dlib==19.16.0