added: license and environment.yml

2026-07-17 19:00:09 +00:00
parent c344983320
commit d34138ec6f
2 changed files with 31 additions and 0 deletions
--- a/environment.yml
+++ b/environment.yml
@@ -0,0 +1,24 @@
+name: reclass
+channels:
+  - pytorch
+  - conda-forge
+  - defaults
+dependencies:
+  - python=3.11
+  - pytorch
+  # 12.4 is recommended. Requires an NVIDIA driver that supports CUDA 12.4 or newer.
+  # if unsure check by running nvidia-smi and see https://pytorch.org/get-started/locally/
+  # and https://anaconda.org/channels/pytorch/packages/pytorch-cuda/overview
+  # - pytorch-cuda=[fill in CUDA toolkit version]
+  - pytorch-cuda=12.4 # default
+  - tensorboard
+  - pandas
+  - scikit-learn
+  - matplotlib
+  - seaborn
+  - numpy
+  - tqdm
+  - jupyter
+  - pip
+  - pip:
+      - transformers