Poetry 如何安装特定版本的 Pytorch

问题原因

pytorch 需要根据 cuda 版本安装特定的版本。

问题分析

pytorch 官方给出的 pip 安装命令如下：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

虽然 poetry 底层的包管理器也是 pip，但是我们依然希望使用 poetry 自己的命令去安装。

假如直接使用标准的安装方式：

poetry add torch torchvision torchaudio

它无法做到让我们指定版本，而每一个人的硬件和驱动可能不太相同，使用的 torch 版本显然是不一致的。

解决方案

这个时候，可以利用 poetry 的 Repositories 特性

1. 首先，我们添加一个 source

   poetry source add --priority=supplemental pytorch_cu121 https://download.pytorch.org/whl/cu121
   

2. 通过指定 source 安装

   poetry add --source pytorch_cu121 torch torchvision torchaudio