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