« BackUsing uv with PyTorchdocs.astral.shSubmitted by charliermarsh 4 hours ago
  • gdiamos 2 hours ago

    uv significantly speeds up my pytorch in docker builds

      # Setup virtual env
      ENV VIRTUAL_ENV=/app/.venv
      ENV PATH="$VIRTUAL_ENV/bin:$PATH"
      RUN python3 -m venv $VIRTUAL_ENV
      RUN . $VIRTUAL_ENV/bin/activate
    
      # install using uv
      RUN pip install uv
      RUN uv pip install torch==${TORCH_VERSION} --index-url https://download.pytorch.org/whl/cpu
    
    The index-url makes it really convenient.
  • minimaxir 3 hours ago

    So uv caused a bit of an issue with me installing PyTorch over the weekend.

    When installed with brew on my MacBook, uv currently has PyTorch 3.13 as a dependency, which is fine. But PyTorch does not currently have a stable wheel that's compatable with Python 3.13! This resulted in very confusing errors. (Solution was to point to the Nightly index)

    That's technically PyTorch's fault, but it's indicitave why a specific page on installing PyTorch is necessary, and it's good to know the documentation specifically calls it out.

    • 0cf8612b2e1e 2 hours ago

      I have run into multiple package problems with 3.13 with a non obvious root cause error message. Thankfully, uv makes it trivial to switch out to 3.12

      • zanie 2 hours ago

        I work on our error messages, feel free to open an issue and we'll do our best to make it clearer