diff --git a/framework.yml b/framework.yml
index f5041ebddcaecae139c7c1aecc75c2aa5cd7708b..1d4f5a0def7769ae21d946628a290bea2927b2e5 100644
--- a/framework.yml
+++ b/framework.yml
@@ -1,4 +1,4 @@
 ---
  Conda 4.10.1
  Python 3.8.5
- tensorflow 2.5.0
\ No newline at end of file
+ tensorflow-gpu 2.4.1
\ No newline at end of file