diff --git a/requirements.txt b/requirements.txt index 09f8f8c..3c177e9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -19,7 +19,7 @@ https://s3-us-west-2.amazonaws.com/ai2-s2-scispacy/releases/v0.2.4/en_core_sci_m # Requirements for the Keras backend, with python 2 # For Vector set: LD_LIBRARY_PATH=/pkgs/cuda-9.0/lib64:/pkgs/cudnn-9.2-v7.1/lib64:$LD_LIBRARY_PATH wheel -tensorflow-gpu==1.5.0 +tensorflow-gpu==2.12.0 h5py==2.10.0 cython==0.29.17 #Original Requirements diff --git a/requirements_py2_mmvenv.txt b/requirements_py2_mmvenv.txt index e27f567..cdf3d4c 100644 --- a/requirements_py2_mmvenv.txt +++ b/requirements_py2_mmvenv.txt @@ -19,7 +19,7 @@ PyYAML==5.3.1 scikit-learn==0.18.2 scipy==1.2.3 six==1.14.0 -tensorflow-gpu==1.5.0 +tensorflow-gpu==2.12.0 tensorflow-tensorboard==1.5.1 tqdm==4.46.0 Werkzeug==1.0.1