如何在environment.yml中指定pip的--extra-index-url参数？

方法1：直接在environment.yml中定义pip依赖与索引源 #

你可以在environment.yml的pip字段下，通过--extra-index-url参数指定PyTorch的CUDA 11.6专属索引源，同时声明torch的具体版本。完整配置示例如下：

name: my-torch-env
channels:
  - defaults
dependencies:
  - python=3.8
  # 可添加其他conda管理的依赖包
  - numpy=1.23.5
  - pip:
    - --extra-index-url https://download.pytorch.org/whl/cu116
    - torch==1.12.1+cu116

创建环境时执行常规命令即可：

conda env create -f environment.yml

方法2：创建环境时通过环境变量指定额外索引源 #

如果不想把索引源写死在配置文件里，可在执行conda env create时，临时设置PIP_EXTRA_INDEX_URL环境变量来指定PyTorch的索引源：

Linux/macOS： #

PIP_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cu116 conda env create -f environment.yml

Windows（PowerShell）： #

$env:PIP_EXTRA_INDEX_URL="https://download.pytorch.org/whl/cu116"; conda env create -f environment.yml

对应的environment.yml只需在pip部分声明torch版本：

name: my-torch-env
channels:
  - defaults
dependencies:
  - python=3.8
  - numpy=1.23.5
  - pip:
    - torch==1.12.1+cu116

内容的提问来源于stack exchange，提问作者Foobar