Foobar
Foobar

Reputation: 8467

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

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.

Upvotes: 17

Views: 15667

Answers (1)

Romain
Romain

Reputation: 21888

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

Upvotes: 27

Related Questions