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