name: hf-train-cu118
channels: [pytorch, nvidia, conda-forge]
dependencies:
  - python=3.10
  - pip
  # ---- Torch 栈：2.1.2 + cu118 ----
  - pytorch=2.1.2
  - torchvision=0.16.2
  - torchaudio=2.1.2
  - pytorch-cuda=11.8

  # ---- 数值栈 ----
  - numpy=1.26.*
  - pandas
  - scipy
  - pyarrow

  # ---- HF 主栈 ----
  - transformers>=4.40,<5
  - accelerate>=0.30,<1.10
  - datasets>=2.18
  - evaluate>=0.4
  - safetensors==0.4.3
  - sentencepiece>=0.1.99
  - tokenizers=0.19.*

  # ---- 依赖/工具 ----
  - protobuf<5
  - huggingface_hub>=0.23
  - tqdm>=4.66
  - scikit-learn>=1.4
  - tensorboard>=2.16
  - packaging
  - regex
  - pyyaml
  - requests
  - fsspec
  - dill
  - multiprocess
  - xxhash
  - aiohttp
  - psutil
  # 可选（按需）
  # - einops
  # - ninja
  # - cmake
  # - xformers==0.0.22.post3  # 若真需要
  # - fastapi                # 若要配合 uvicorn 跑服务