PyTorch 是一款众所周知的安装棘手的 Python 软件,因为需要为不同的 Python 版本和 GPU 加速器组合(例如不同的 CUDA 版本)提供单独的轮子。
uv 现在有 PyTorch 的专用文档,我发现它非常有用 – 它清楚地解释了挑战,然后准确地显示了如何配置pyproject.toml
,以便uv
知道应该从哪里安装每个包的哪个版本。
标签:包装, python , uv , pytorch , pip
原文: https://simonwillison.net/2024/Nov/19/using-uv-with-pytorch/#atom-everything