跳转至

通过uv安装PyTorch的实践

PyTorch是时下进行深度学习模型训练的主流框架,其安装过程也相对复杂。传统的基于conda的虚拟环境配置模式因为种种弊端,使得越来越多的开发者逐渐开始寻找更为现代化的管理方式。而近年来兴起的uv为我们提供了一套不错的解决方案。

创建项目

先创建一个项目来管理所需的依赖:

Bash
uv init -p 3.8

安装PyTorch

默认使用外国的镜像源会失败,手动修改为国内的镜像源。

Bash
uv add torch==2.4.1 torchvision==0.19.1 --default-index https://pypi.tuna.tsinghua.edu.cn/simple

注意

在安装时,python版本、torch版本和torchvision版本必须符合官方的对应标准,否则安装会出错。 对应关系参考博客:深度学习 | pytorch + torchvision + python 版本对应及环境安装