【Python】UV:境内的深度学习环境搭建

本文将介绍如何在国内网络环境中使用 UV 快速构建深度学习开发环境,包括 UV 的安装、Python 下载加速、项目初始化、依赖管理与运行方式。


一、安装 UV 工具

在国外网络环境下可以直接通过官方脚本安装:

sh 复制代码
curl -LsSf https://astral.sh/uv/install.sh | sh

在国内环境中建议使用更稳定的方式安装:

1. Ubuntu

Ubuntu 用户可通过 Snap 直接安装国内镜像版本:

sh 复制代码
sudo snap install astral-uv --classic

2. macOS

macOS 用户可使用 Homebrew,在配置好国内镜像源后执行:

sh 复制代码
brew install uv

3. 通过 pip 安装

若系统已有 pip,可以从国内 PyPI 镜像安装:

sh 复制代码
pip install uv -i https://pypi.tuna.tsinghua.edu.cn/simple

二、使用 UV 安装 Python

由于默认下载源 Github 经常受到地域限制,UV 支持通过环境变量指定国内镜像以加速 Python 下载。

1. 设置环境变量

Windows:

cmd 复制代码
set UV_PYTHON_INSTALL_MIRROR=https://registry.npmmirror.com/-/binary/python-build-standalone

Unix / Linux / macOS:

sh 复制代码
export UV_PYTHON_INSTALL_MIRROR=https://registry.npmmirror.com/-/binary/python-build-standalone

2. 安装指定版本 Python

例如安装 Python 3.12:

sh 复制代码
uv python install 3.12

UV 将自动从镜像站点下载并安装对应的 Python 版本,适合国内环境使用。


三、初始化项目环境

1. 创建配置文件

进入你的工作目录,例如 deeplearn/,然后执行:

sh 复制代码
uv init --bare

该命令会创建由pyproject.toml定义的最小化 Python 项目结构,包括:

toml 复制代码
[project]
name = "deeplearn"
version = "0.1.0"
requires-python = ">=3.12"
dependencies = []

2. 添加默认 PyPI 镜像源

为进一步提升依赖安装速度,可在 pyproject.toml 中追加:

toml 复制代码
[[tool.uv.index]]
url = "https://pypi.tuna.tsinghua.edu.cn/simple"

3. 配置 PyTorch 官方镜像

toml 复制代码
[[tool.uv.index]]
name = "pytorch-cu130"
url = "https://download.pytorch.org/whl/cu130"
explicit = true

[tool.uv.sources]
torch = { index = "pytorch-cu130" }
torchvision = { index = "pytorch-cu130" }

这样即可确保 PyTorch 和 TorchVision 从正确的 CUDA 对应版本安装。


四、使用 UV 进行包管理

1. 安装依赖

UV 使用 uv add 安装包,并自动写入 pyproject.toml

sh 复制代码
uv add torch torchvision

2. 删除依赖

sh 复制代码
uv remove 包名

3. 同步环境

当你从其他机器复制了项目,需要恢复环境时:

sh 复制代码
uv sync

五、运行程序

使用 uv run 可在虚拟环境中运行 Python 程序,无需手动激活环境:

sh 复制代码
uv run main.py

UV 会自动创建 .venv 虚拟环境,若需要手动激活 .venv

  • Unix / Linux / macOS:

    sh 复制代码
    source .venv/bin/activate
  • Windows:

    powershell 复制代码
    .venv\Scripts\activate
相关推荐
大公产经晚间消息9 分钟前
美团医药健康与鱼跃、海氏海诺等头部医疗器械品牌深化合作,开拓即时零售新主场
人工智能
xianluohuanxiang13 分钟前
高精度气象:极端天气一来,零售最先出问题的不是客流,而是补货体系和损失控制
开发语言·人工智能·深度学习·机器学习·零售
科威舟的代码笔记20 分钟前
中国专供AI社区腾讯SkillHub来了!
人工智能·ai大模型·skillhub
贵州数擎科技有限公司25 分钟前
NumPy 从数组操作理解深度学习的计算本质
人工智能·numpy
山海AI手册28 分钟前
030、AI应用前端展示:Streamlit快速构建交互式Web应用
前端·人工智能
菱玖29 分钟前
RAG 技术详解
人工智能·语言模型·aigc
d1z88831 分钟前
(十八)32天GPU测试从入门到精通-TensorRT-LLM 部署与优化day16
人工智能·python·深度学习·gpu·tensorrt
前端摸鱼匠33 分钟前
YOLOv11 在零售领域实战:利用公开的商品检测数据集 (如 SKU110K 的子集),训练一个 YOLOv11 模型,用于识别货架上的各种商品
人工智能·yolo·目标检测·ai·目标跟踪·视觉检测·零售
imbackneverdie34 分钟前
分享一些高级感科研绘图配色
图像处理·人工智能·ai·aigc·ai绘画·贴图·科研绘图
行者-全栈开发34 分钟前
AI 驱动的智能行程规划系统:腾讯地图 Map Skills 实战
人工智能·路径规划·ai agent·多人协同·tool calling·mcp 协议·poi 检索