成功了！|| Poetry安装pytorch || 整理自github项目Poetry下的issue

原创

已于 2023-08-04 17:04:18 修改 · 3.6k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #issue #人工智能

于 2023-08-04 16:27:45 首次发布

文章讲述了在使用Poetry管理PyTorch时遇到的问题，如版本匹配、CUDA寻找和镜像加速。提供了选择特定版本和设置次级源的方法，但作者认为目前poetry对torch的管理功能有限，推荐使用pip进行环境管理。

在使用Poetry安装pytorch的时候，常常会遇到各种问题：首先是使用add添加时，会说只有torch没有什么pytorch，很显然，它是直接针对包的，第二点是，如果是一台没有nvidia显卡的机器，由于poetry没有找到适配的cuda，它会从最高版本开始一遍遍地尝试安装cuda，即使添加了清华镜像也还是慢得可怜，最重要的是，不知道在终点会不会是一个“错误”。因此，在poetry的项目下面也有人针对这个问题展开了讨论，下面是一个比较详尽的使用poetry安装的方式。

Instructions for installing PyTorch #6409

Instructions for installing PyTorch · Issue #6409 · python-poetry/poetry · GitHub

选项一：选择一个你需要的特定版本

你需要选择所需的特定wheel。这个网址https://download.pytorch.org/whl/torch_stable.html里面选。例如，如果您想要CUDA 11.6、Python 3.10和Windows，请在该页面上搜索cu116-cp310-cp310-win_amd64.whl以查看torch、torchhave和torchvision的匹配项。

那么你写的pyproject,toml应该长这样

[tool.poetry.dependencies]
python = "^3.10"
numpy = "^1.23.2"
torch = { url = "https://download.pytorch.org/whl/cu116/torch-1.12.1%2Bcu116-cp310-cp310-win_a