name: KPMBert
channels:
  - defaults
  - conda-forge
  - nvidia
  - pytorch
dependencies:
  - python=3.11
  - cudatoolkit=12.4
  - pip
  - pip:
      - tensorflow==2.18.0
      - numpy>=1.17
      - pandas
      - transformers==4.48.3
      - scikit-learn
      - matplotlib
      - tqdm
      - torch==2.5.1+cu124
      - torchvision==0.20.1+cu124
      - pytorch-crf==0.7.2
      - keras