尽管这个问题已经有点老了,但它在我的谷歌搜索中排名很高。因为我认为我找到了一个非常有用的解决这个问题的方法,所以我想在这里分享它:
由于 Poetry 仍然无法解析 pytorch+cuda 包,因此您必须指定其特定轮子的路径。由于这打破了我在某些平台上需要 Cuda 支持而仅在其他平台上需要 CPU 的用例,所以我最终得到了这个规范:
torch = [
{url="https://download.pytorch.org/whl/cu113/torch-1.11.0%2Bcu113-cp310-cp310-linux_x86_64.whl",markers = "sys_platform == 'linux'"},
{version="^1.11.0",markers = "sys_platform == 'darwin'"}]
torchvision = [
{url="https://download.pytorch.org/whl/cu113/torchvision-0.12.0%2Bcu113-cp310-cp310-linux_x86_64.whl",markers = "sys_platform == 'linux'"},
{version="^0.12.0",markers = "sys_platform == 'darwin'"}]
torchaudio = [
{url="https://download.pytorch.org/whl/cu113/torchaudio-0.11.0%2Bcu113-cp310-cp310-linux_x86_64.whl",markers = "sys_platform == 'linux'"},
{version="^0.11.0",markers = "sys_platform == 'darwin'"}]
希望这会有所帮助,干杯!