How to specify pip –extra-index-url in environment.yml?

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.

Asked By: Foobar

||

Answers:

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

Answered By: Romain
Categories: questions Tags: , ,
Answers are sorted by their score. The answer accepted by the question owner as the best is marked with
at the top-right corner.