Poetry Pytorch
Contents
问题
最近试了试 yolo11，直接安装
poetry add ultralytics
这样下来的是 cpu 版本的 torch，在执行下面的代码时，会报torch编译时没有使能cuda
model = YOLO("yolo11n.pt")
model.to("cuda")
解决方案
解决方案在这里
首先要安装cuda toolkit，然后查看下版本，最后在这里看下pip index应该用什么url， 最后执行下面的命令即可，torch/torchvision/torchaudio 都会被替换为 cuda 版本。
poetry source add --priority=explicit pytorch https://download.pytorch.org/whl/cu124
poetry add torch torchvision torchaudio --source pytorch
安装完成后，可以用如下代码检查是否安装成功
import torch
print(torch.cuda.is_available())
print("CUDA Version:", torch.version.cuda)
其他
这里说的，阿里的pytorch wheels， 是不能用的，因为它没有提供完整的index服务，所以poetry没法直接用，会报找不到package