[Python] How to specify a nightly Pytorch version in environment.yml

Consider the following environment.yml file:

channels:
- nvidia
- defaults
- conda-forge
dependencies:
- bottleneck>=1.3.6
- catboost>=0.24.4
...
- pip:
- xgboost==2.1.4
...


How do I add the following pip command to the yml file without disrupting the current behavior of the other pip install commands above?

pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128

Continue reading...