diff --git a/environments/finetuneanon.yml b/environments/finetuneanon.yml
index 7a23b7a2..4e1b4c75 100644
--- a/environments/finetuneanon.yml
+++ b/environments/finetuneanon.yml
@@ -7,8 +7,9 @@ dependencies:
   - colorama
   - flask-socketio
   - pytorch
+  - cudatoolkit=11.1
   - tensorflow-gpu
-  - python=3.8.*
+  - python=3.9.*
   - pip
   - git
   - pip:
diff --git a/environments/huggingface.yml b/environments/huggingface.yml
index 659ecc74..93398e31 100644
--- a/environments/huggingface.yml
+++ b/environments/huggingface.yml
@@ -8,6 +8,8 @@ dependencies:
   - colorama
   - flask-socketio
   - pytorch
+  - python=3.9.*
+  - cudatoolkit=11.1
   - tensorflow-gpu
   - transformers
   - pip