3 ms·

https://docs.astral.sh/uv/guides/integration/pytorch/#automatic-backend-selection https://docs.astral.sh/uv/guides/integration/pytorch/#automa... doesn't work?

• lz400 1y ago
  the problem is that you still need to install all the low level stuff manually, conda does it automatically
  • gcarvalho 1y ago
    I was pleasantly surprised to try the guide out and see that it just worked: λ uv venv && uv pip install torch --torch-backend=auto λ uv run python -c 'import torch; print(torch.cuda.is_available())' True This is on Debian stable, and I don't remember doing any special setup other than installing the proprietary nvidia driver.
  • pcwelder 1y ago
    This script has been sufficient for me to configure gpu drivers on fresh ubuntu machines. It's just uv add torch after this. https://cloud.google.com/compute/docs/gpus/install-drivers-gpu#install-script https://cloud.google.com/compute/docs/gpus/install-drivers-g... (NOTE: not gcloud specific)