kd_env.yml

name: kd
channels:
  - pytorch
  - defaults
dependencies:
  - python=3.9
  - cudatoolkit
  - pip
  - pytorch
  - torchaudio
  - torchtext
  - torchvision
  - numpy
  - pip:
    - datasets
    - huggingface-hub
    - matplotlib
    - multiprocess
    - pandas
    - pytorch-lightning
    - scikit-learn
    - scipy
    - seaborn
    - transformers
    - sentence-transformers
    - mlflow
    - gensim
    - tdqm
    - spacy
    - nltk
    - jupyter
    - notebook
    - ipython
    - statsmodels
    - xgboost
    - wheel
    - tensorboard
    - sortedcontainers
    - lxml
    - plotly
    - wandb
    - jiwer
    - rouge_score
    - sacrebleu
    - accelerate
    - bert_score
    - experiment-impact-tracker
    - torchprofile