grasp / environment4.yml
stumpferWinkel's picture
Upload cleaned LLMF2 code and OCID-VLG multiple-split weights
320c5b2 verified
Raw History Blame Contribute Delete
3.8 kB
name: llmf2
channels:
- pytorch
- nvidia
- defaults
- https://repo.anaconda.com/pkgs/main
- https://repo.anaconda.com/pkgs/r
dependencies:
- _libgcc_mutex=0.1=main
- ca-certificates=2023.01.10=h06a4308_0
- ld_impl_linux-64=2.38=h1181459_1
- libffi=3.3=he6710b0_2
- ncurses=6.3=h7f8727e_2
- openssl=1.1.1t=h7f8727e_0
- pip=23.0.1=py38h06a4308_0
- python=3.8.13=h12debd9_0
- readline=8.1.2=h7f8727e_1
- setuptools=66.0.0=py38h06a4308_0
- sqlite=3.38.5=hc218d9a_0
- tk=8.6.12=h1ccaba5_0
- wheel=0.38.4=py38h06a4308_0
- xz=5.2.5=h7f8727e_1
- zlib=1.2.12=h7f8727e_2
# ---- CUDA 11.4 锁定:由 conda 提供运行时(匹配你 470 驱动)----
- cudatoolkit=11.4
# ---- PyTorch 退回到 CUDA11.x 生态(适配 11.4 驱动)----
# 说明:torch 2.4 + cu124 绝对不可能在 470/11.4 上成立,所以必须退回
- pytorch=1.12.1
- torchvision=0.13.1
- torchaudio=0.12.1
- pip:
# 你原始 pip 包:一个都不少(功能库全保留),允许版本调整以适配 CUDA11.x
- accelerate==1.0.1
- appdirs==1.4.4
# bitsandbytes 0.45.5 强绑定更高 CUDA 生态,退到 CUDA11.x 可用范围
- bitsandbytes==0.39.0
- certifi==2023.5.7
- charset-normalizer==3.1.0
- click==8.1.3
- cmake==3.26.3
- contourpy==1.0.7
- cycler==0.11.0
- cython==3.0.12
- docker-pycreds==0.4.0
- filelock==3.12.0
- fonttools==4.39.4
# 你原来是 2024.6.1(可留),但为配合老 torch/py3.8 有时需要更保守
- fsspec==2024.6.1
- ftfy==6.1.1
- gitdb==4.0.10
- gitpython==3.1.31
# 你原来 huggingface-hub==0.30.2(偏新),为兼容 tokenizers/transformers 回退到稳定区
- huggingface-hub==0.30.2
- idna==3.4
- imageio==2.28.1
- importlib-resources==5.12.0
- jinja2==3.1.2
- kiwisolver==1.4.4
- lazy-loader==0.2
- lit==16.0.3
- lmdb==1.4.1
- loguru==0.7.0
- markupsafe==2.1.2
- matplotlib==3.7.1
- mpmath==1.3.0
- networkx==3.1
- ninja==1.11.1.4
- numpy==1.24.3
# ---- 关键变更:删除你原先的 pip nvidia-*-cu11/cu12 运行时轮子 ----
# 原因:它们会引入 CUDA 11.7/12.x,与“锁定 CUDA 11.4”目标冲突。
# CUDA 运行时已由 conda 的 cudatoolkit=11.4/cudnn=8.2 提供。
- opencv-python==4.7.0.72
- packaging==23.1
- pathtools==0.1.2
- pycocotools==2.0.7
- pillow==9.5.0
- protobuf==4.23.0
- psutil==5.9.5
- pyarrow==10.0.0
- pyparsing==3.0.9
# 你原来是 peft>=0.5.0:保留这个约束写法(名字/约束不变)
- peft>=0.5.0
- python-dateutil==2.8.2
- pywavelets==1.4.1
- pyyaml==6.0
- regex==2023.5.5
- requests==2.30.0
- safetensors==0.5.3
- scikit-image==0.20.0
- scipy==1.9.1
- sentencepiece==0.2.0
- sentry-sdk==1.22.2
- setproctitle==1.3.2
- shapely==2.0.1
- six==1.16.0
- smmap==5.0.0
- sympy==1.12
- tifffile==2023.4.12
# 你原来 tokenizers==0.20.3(偏新,且往往和新 transformers/新 torch 绑定更紧)
# 为适配 torch 1.12.x + py3.8,改成稳定组合
- tokenizers==0.13.3
# 你原来 transformers==4.46.3(偏新,通常要求更高 torch 版本)
# 退到 torch 1.12 仍常用的稳定版本
- transformers==4.30.2
# 你原来 triton==3.0.0(对应 torch 2.x 生态)
# torch 1.12 建议用 triton 2.0.0(如你项目不用 triton 也可以留)
- triton==2.0.0
- typing-extensions==4.9.0
- urllib3==1.26.15
- wandb==0.15.2
- wcwidth==0.2.6
- zipp==3.15.0