algo: introduce conda env file

Signed-off-by: Zach Zhu <[email protected]>
traas-stack · Oct 22, 2023 · 47f051e · 47f051e
1 parent 7c9ea6a
commit 47f051e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/algorithm/environment.yml b/algorithm/environment.yml
@@ -0,0 +1,11 @@
+name: kapacity
+dependencies:
+  - python=3.11.*
+  - pytorch::pytorch=2.0.*
+  - pandas=1.5.*
+  - pyyaml=6.0.*
+  - scikit-learn=1.2.*
+  - lightgbm=3.3.*
+  - protobuf=3.20.*
+  - grpcio=1.48.*
+  - python-kubernetes=23.6.*