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.* ----
  - numpy=1.26.*
  # ---- 常用科学/系统库 ----
  - pandas
  - scipy
  - pyarrow
  - uvicorn
  - git
  # ---- HF 主栈 + 其运行时依赖（全部走 conda，不让 pip 动依赖）----
  - transformers>=4.40
  - accelerate>=0.30
  - datasets>=2.18
  - evaluate>=0.4
  - safetensors>=0.4
  - sentencepiece>=0.1.99
  - tokenizers=0.19.*
  - huggingface_hub>=0.23
  - tqdm>=4.66
  - scikit-learn>=1.4
  - tensorboard>=2.16
  - packaging
  - regex
  - pyyaml
  - requests
  - fsspec
  - dill
  - multiprocess
  - xxhash
  - aiohttp
  - psutil