diff --git a/requirements-colab.txt b/requirements-colab.txt index 5f2216f..6e1aa44 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==1.15.2 plaidml-keras==0.5.0 scikit-image tqdm diff --git a/requirements-cuda.txt b/requirements-cuda.txt index cbb3279..c1f04a1 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==1.15.2 plaidml==0.6.0rc1 plaidml-keras==0.5.0 scikit-image