通过uv安装PyTorch的实践
PyTorch是时下进行深度学习模型训练的主流框架,其安装过程也相对复杂。传统的基于conda的虚拟环境配置模式因为种种弊端,使得越来越多的开发者逐渐开始寻找更为现代化的管理方式。而近年来兴起的uv为我们提供了一套不错的解决方案。
创建项目
先创建一个项目来管理所需的依赖:
Bash | |
---|---|
安装PyTorch
默认使用外国的镜像源会失败,手动修改为国内的镜像源。
Bash | |
---|---|
注意
在安装时,python版本、torch版本和torchvision版本必须符合官方的对应标准,否则安装会出错。 对应关系参考博客:深度学习 | pytorch + torchvision + python 版本对应及环境安装