top | item 43351341

(no title)

ibic | 11 months ago

I'm not sure if I got your issue, but I can do platform-dependent `index` `pytorch` installation using the following snippet in `pyproject.toml` and `uv sync` just handles it accordingly.

[tool.uv.sources] torch = [{ index = "pytorch-cu124", marker = "sys_platform == 'win32'" }]

discuss

order

satvikpendem|11 months ago

Some Windows machines have compatible GPUs while others don't, so this doesn't necessarily help. What is really required is querying the OS for what type of compute unit it has and then installing the right version of an ML library, but I'm not sure that will be done.

fluidcruft|11 months ago

Even without query, just setting an environment variable or having remember which extras are already applied to the already synced .venv some way.