On: Using uv with PyTorch

[gdiamos]

Nice, I wonder if there is a way to make it conditional, e.g. to pick a different key for a cpu vs cuda build.

[kristjansson → gdiamos]

That’s basically what TFA is about? Set up multiple index urls, and use a bog-standard platform marker on the PyTorch dep to pick between them.

[ttyprintk → gdiamos]

Eh, in that case the environment variable is a bit more evident in a Dockerfile.