@XnneHang 在 使用 uv 配置为依赖于 Pytorch 的 pyproject 单独配置cpu或cuda版本. 中发帖
Refrence:
https://docs.astral.sh/uv/guides/integration/pytorch/#installing-pytorch
ps . uv 官方真的太贴心了,这是喂饭级的,把各种cpu,cu11.2,cu12,intel gpu…的基础配置都写出来了。
我印象里面看到poetry用户在安装的时候考虑用指定whl路径的方法,但是存在很大局限性,因为 wh l是锁死 python 版本和系统的。
Installing a specific PyTorch build (f/e CPU-only) with Poetry
问题描述:
正常情况下,我们使用 uv add torch==2.1.0时,安装的是 cpu+cuda 版本的 torch:
xnne@xnne-PC:~/code/Auto_Caption_Generated_Off...