python包管理rye安装torch

发布时间:2023年12月20日

最近在使用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
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。