diff --git a/requirements-hpc.txt b/requirements-hpc.txt
index 58b130e..2f98cb9 100644
--- a/requirements-hpc.txt
+++ b/requirements-hpc.txt
@@ -6,7 +6,7 @@
 -e .
 
 # external packages
-tensorflow-gpu==2.1.0
+tensorflow-gpu==2.6.4
 tensorboard==2.1.0
 h5py==2.10.0
 biopython==1.76
diff --git a/requirements.txt b/requirements.txt
index 85fbd48..dae7a89 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@
 -e .
 
 # external packages
-tensorflow-gpu==2.1.0
+tensorflow-gpu==2.6.4
 tensorboard==2.1.0
 h5py==2.10.0
 biopython==1.76