python包管理rye安装torch

2023-12-20 08:38:42

最近在使用pip或中poetry对python进行包管理，始终觉得不是很理想。2023年flask的作者参考rust的管理哲学，编写了rye的python包管理工具，故试用之。使用过程中，遇到安装torch的问题，故这里写一下笔记。

1. 本地windows安装torch（国内更推荐）

到官网找到自己需要的torch版本进行下载：
https://download.pytorch.org/whl/torch/
直接下载:
https://download.pytorch.org/whl/cu121/torch-2.1.0%2Bcu121-cp310-cp310-win_amd64.whl
这里下载到G:/download/python_whl目录后w使用rye相关指令。

rye add "torch==2.1.0+cu121" --path "G:/download/pyton_whl"
rye sync --verbose

2. 网络安装torch(国内容易超时)

配置源,修改pyproject.toml文件

[[tool.rye.sources]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu121"

安装指定版本的torch

rye add "torch==2.1.0+cu121"
rye sync --verbose

文章来源:https://blog.csdn.net/cookic12346/article/details/135080609
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。