Reputation: 8467
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
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