如何通过Poetry从含.whl的自定义URL安装Python包(含信任主机)
从requirements.txt迁移PyTorch依赖到Poetry的正确方式
你的需求对应把requirements.txt里的--trusted-host和-f(额外索引源)转换为Poetry的配置,同时保留依赖版本的跨环境兼容性,具体步骤如下:
1. 配置PyTorch CUDA 113的专属索引源
首先把原来的-f https://download.pytorch.org/whl/cu113/torch_stable.html和--trusted-host download.pytorch.org转换为Poetry的仓库配置:
# 添加自定义仓库并命名为pytorch-cu113 poetry config repositories.pytorch-cu113 https://download.pytorch.org/whl/cu113 # 标记该仓库为可信,对应原requirements的--trusted-host poetry config repositories.pytorch-cu113.trusted true
2. 添加指定版本的PyTorch依赖
使用--source参数指定从刚才配置的仓库拉取依赖,Poetry会自动匹配当前环境(Python版本、操作系统、架构)对应的wheel文件,解决你第一种方法的兼容性问题:
poetry add torch==1.10.1+cu113 torchvision==0.11.2 --source pytorch-cu113
手动修改pyproject.toml的方式(可选)
如果需要直接编辑配置文件,在pyproject.toml中添加以下内容:
# 配置自定义仓库 [tool.poetry.repositories] pytorch-cu113 = { url = "https://download.pytorch.org/whl/cu113", trusted = true } # 添加依赖并指定来源 [tool.poetry.dependencies] # 其他依赖... torch = { version = "1.10.1+cu113", source = "pytorch-cu113" } torchvision = { version = "0.11.2", source = "pytorch-cu113" }
完成后运行poetry install即可完成依赖安装,效果和原requirements.txt完全一致,且支持跨环境自动适配。
内容的提问来源于stack exchange,提问作者Or Shemtov
相关产品推荐
相关产品推荐

