7

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.

Romain
  • 19,910
  • 6
  • 56
  • 65
Foobar
  • 7,458
  • 16
  • 81
  • 161

1 Answers1

15

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

Romain
  • 19,910
  • 6
  • 56
  • 65