diff --git a/AttentiveChrome/model-template.yaml b/AttentiveChrome/model-template.yaml
index c6e6fdabe..0531a60a2 100755
--- a/AttentiveChrome/model-template.yaml
+++ b/AttentiveChrome/model-template.yaml
@@ -31,10 +31,11 @@ info: # General information about the model
         - RNA expression
 dependencies:
     conda: # install via conda
-      - python==3.6.3
-      - numpy  
-      - pytorch::pytorch-cpu 
-      - pytorch::torchvision-cpu   
+      - python=3.6.3
+      - numpy=1.19.2
+      - pytorch::pytorch-cpu=1.3.1
+      - pytorch::torchvision-cpu=0.3.0
+      - pip=21.0.1  
 schema:  # Model schema
     inputs:
         shape: (100, 5)  # array shape of a single sample (omitting the batch dimension)