diff --git a/requirements-colab.txt b/requirements-colab.txt index ccdf38b..27f41cd 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==1.15.0 plaidml-keras==0.5.0 scikit-image tqdm diff --git a/requirements-cuda.txt b/requirements-cuda.txt index edfa576..790b846 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==1.15.0 plaidml==0.6.0 plaidml-keras==0.5.0 scikit-image