diff --git a/requirements-gpu.txt b/requirements-gpu.txt index 4e3bc4f..d10e437 100644 --- a/requirements-gpu.txt +++ b/requirements-gpu.txt @@ -3,6 +3,6 @@ numpy scipy==1.0.0 matplotlib scikit-image -tensorflow-gpu==1.12.0 # GPU version of TensorFlow. +tensorflow-gpu==2.6.4 # GPU version of TensorFlow. imreg_dft pytest