45 lines
832 B
YAML
45 lines
832 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.* ----
|
||
- 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
|
||
|