Update paynt-learner.dockerfile

Additional dependencies with specific versions
randriu · Dec 4, 2023 · d184faa · d184faa
1 parent 7da363f
commit d184faa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paynt-learner.dockerfile b/paynt-learner.dockerfile
@@ -1,6 +1,6 @@
 ARG paynt_base=randriu/paynt:ci
 FROM $paynt_base
 
-RUN pip install torch
+RUN pip install torch==2.1.*
 
-RUN pip install ipykernel joblib tensorboard einops
+RUN pip install ipykernel joblib tensorboard==2.15.* einops==0.7.* gym==0.22.* pygame==2.5.*