【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
相关推荐
七宝大爷25 分钟前
基于人类反馈的强化学习(RLHF):ChatGPT“对齐”人类的秘密武器
人工智能·chatgpt
shayudiandian27 分钟前
ChatGPT风格对话机器人搭建教程
人工智能·chatgpt·机器人
腾讯云开发者29 分钟前
TVP首场香港活动重磅启幕,AI出海变革风向如何把握?
人工智能
wasp52029 分钟前
Spring AI 代码分析(十)--Spring Boot集成
人工智能·spring boot·spring
AI即插即用36 分钟前
即插即用系列 | 2025 MambaNeXt-YOLO 炸裂登场!YOLO 激吻 Mamba,打造实时检测新霸主
人工智能·pytorch·深度学习·yolo·目标检测·计算机视觉·视觉检测
BagMM1 小时前
DetLH论文阅读
人工智能·计算机视觉·目标跟踪
Luminbox紫创测控1 小时前
太阳光模拟关于UV紫外老化测试的七个关键问题
uv
电棍2331 小时前
uv配置cosmos transfer环境时候的报错记录
uv
fundroid1 小时前
Androidify:谷歌官方 AI + Android 开源示例应用
android·人工智能·开源