diff --git a/requirements-colab.txt b/requirements-colab.txt index 6ed6edc..4e30607 100644 --- a/requirements-colab.txt +++ b/requirements-colab.txt @@ -2,7 +2,7 @@ numpy==1.17.0 h5py==2.9.0 Keras==2.2.4 opencv-python==4.1.0.25 -tensorflow-gpu==1.13.1 +tensorflow-gpu==2.9.3 plaidml-keras==0.5.0 scikit-image tqdm diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 0d6ab03..471a27b 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -2,7 +2,7 @@ numpy==1.17.0 h5py==2.9.0 Keras==2.2.4 opencv-python==4.1.0.25 -tensorflow-gpu==1.12.0 +tensorflow-gpu==2.9.3 plaidml==0.6.0 plaidml-keras==0.5.0 scikit-image