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

Question

Conda can create an environment.yml that specifies both conda packages & pip packages. The problem is, I want to specify a pip package ( torch==1.12.1+cu116 ), that is only available in the following index: https://download.pytorch.org/whl/cu116 .

How can I specify this in the environment.yml? Or at the very least, when running conda env create -f environment.yml , I would like to specify the extra index for pip.

Answer 1

This configuration should work, see the advanced-pip-example for other options.

name: foo
channels:
  - defaults
dependencies:
  - python
  - pip
  - pip:
    - --extra-index-url https://download.pytorch.org/whl/cu116
    - torch==1.12.1+cu116

See also

• Combining conda environment.yml with pip requirements.txt
• Can conda be configured to use a private pypi repo?