diff --git a/requirements-colab.txt b/requirements-colab.txt index 5f2216f..213a588 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.9.3 plaidml-keras==0.5.0 scikit-image tqdm diff --git a/requirements-cuda.txt b/requirements-cuda.txt index cbb3279..f867bbc 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.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.12.0 +tensorflow-gpu==2.9.3 plaidml==0.6.0rc1 plaidml-keras==0.5.0 scikit-image