What is the pixi equilvalent to pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu?

[asmith26]

Hi, I'm trying to install a CPU version of (py)torch using the following command, but I can't seem to get pixi to install torch from https://download.pytorch.org/whl/cpu.

What is the pixi equilvalent to the following (I'm using a pyproject.toml file)?

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

Many thanks for any help!

[asmith26]

I've got it working:

Option 1 (easier)

Following https://pixi.sh/v0.22.0/reference/configuration/#the-pypi-options-table, add to pyproject.toml:

[tool.pixi.pypi-options]
index-url = "https://pypi.org/simple"
extra-index-urls = ["https://download.pytorch.org/whl/cpu"]

Option 2 (more work)

Following https://pixi.sh/v0.22.0/reference/configuration/#pypi-dependencies, pass in the appropriate URL (within pyproject.toml, e.g.

torch = { url = "https://download.pytorch.org/whl/cpu/torch-2.4.0%2Bcpu-cp312-cp312-linux_x86_64.whl#sha256=78dbf5f2789933a7ea2dabeead4daa44679b1e0d8eb35ddb7071c8ab7b181eb3" }

---

Thanks again for an amazing lib!

[m0n203]

I am using this way:

pixi.toml

[pypi-dependencies]
torch = { version = "*", index = "https://download.pytorch.org/whl/cpu" }

but it's no way to do this w/ pixi add --pypi, have to edit toml manually.