如何在environment.yml中配置PyTorch Nightly版本且不影响现有依赖安装

别担心，这个操作超简单，完全不会打乱你现有的依赖安装流程。你只需要把带参数的PyTorch Nightly安装命令转换成conda的environment.yml支持的格式，直接追加到现有的pip依赖列表末尾就行。

具体修改后的完整environment.yml示例如下：

channels:
  - nvidia
  - defaults
  - conda-forge
dependencies:
  - bottleneck>=1.3.6
  - catboost>=0.24.4
  # 这里保留你原来的其他conda依赖
  ...
  - pip:
    - xgboost==2.1.4
    # 这里保留你原来的其他pip依赖
    ...
    # 新增的PyTorch Nightly安装项
    - --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128

为什么这样做没问题？因为conda在处理pip小节的内容时，会严格按列表顺序执行安装：先装完你原来的所有pip依赖，再用指定的--pre（安装预发布版）和自定义索引地址来安装PyTorch Nightly的三个包，完全不会影响前面已有的依赖。

如果你担心参数会意外影响其他包，也可以拆成更明确的单独条目（效果和上面完全一致）：

- pip:
    - xgboost==2.1.4
    ...
    - torch --pre --index-url https://download.pytorch.org/whl/nightly/cu128
    - torchvision --pre --index-url https://download.pytorch.org/whl/nightly/cu128
    - torchaudio --pre --index-url https://download.pytorch.org/whl/nightly/cu128

两种写法都能正常工作，选你看着舒服的就行。

备注：内容来源于stack exchange，提问作者Martin