diff --git a/environment.yml b/environment.yml
index c9e6cf6..526b5c8 100644
--- a/environment.yml
+++ b/environment.yml
@@ -8,8 +8,8 @@ channels:
 dependencies:
   - python=3.9
   - tensorflow-gpu=2.8.0
-  - cudatoolkit=11.0
-  - cudnn=8.9
+  - cudatoolkit=11.2
+  - cudnn=8.1
   - pytorch-gpu
   - igraph
   - numpy