49 lines
892 B
YAML
49 lines
892 B
YAML
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
|
||
- datasets>=2.18
|
||
- evaluate>=0.4
|
||
- safetensors>=0.4
|
||
- 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 跑服务
|