PyTorch 环境搭建与包管理实践
检测 PyTorch 是否已部署
确认当前环境是否包含 PyTorch，可通过多种途径实现：
交互式验证
启动 Python 解释器后，执行以下语句。若环境正常，将返回版本字符串；反之则触发 ModuleNotFoundError。
import torch as th
print(th.__version__)包管理器查询
通过 pip 或 conda 检索已安装包：
# pip 方案
pip list | findstr torch # Windows
pip list | grep torch # Linux/macOS
# conda 方案
conda list "^torch"详细信息展示
pip show torch此命令输出包含版本、安装路径、依赖关系等元数据。
注意：操作前务必激活目标虚拟环境，避免跨环境误检。
PyTorch 安装指南
基于 pip 的安装
PyTorch 官方提供动态命令生成器，根据硬件配置获取精确指令。以下为常见场景示例：
# GPU 加速版（CUDA 12.1）
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
# 纯 CPU 版本
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu基于 conda 的安装
# GPU 版本
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
# CPU 专用版本
conda install pytorch torchvision torchaudio cpuonly -c pytorch关键注意项
- 驱动兼容性：CUDA 版本需与 NVIDIA 驱动匹配，可通过
nvidia-smi查看支持的 CUDA 最高版本 - 隔离环境：推荐采用
venv或conda env创建独立空间，防止依赖冲突 - 权限处理：无管理员权限时，追加
--user参数安装至用户目录 - 网络加速：国内用户可配置清华、阿里云等镜像源提升下载效率
定位 pip 安装路径
通用查询方法
python -m site该命令列出所有 site-packages 路径，其中：
USER_SITE：用户级安装目录（pip install --user）sys.path中的site-packages：系统级或虚拟环境目录
虚拟环境路径获取
先激活环境，再执行查询：
# Windows
.\env_name\Scripts\activate
# Unix/macOS
source env_name/bin/activate
# 验证当前路径
python -c "import site; print(site.getsitepackages())"典型路径结构参考
| 平台 | 类型 | 路径示例 |
|---|---|---|
| Linux | 系统级 | /usr/local/lib/python3.10/site-packages |
| Linux | 用户级 | ~/.local/lib/python3.10/site-packages |
| Windows | 系统级 | C:\Python310\Lib\site-packages |
| Windows | 用户级 | %APPDATA%\Python\Python310\site-packages |
若需精确定位特定包的所在位置，可执行：
python -c "import torch; print(torch.__file__)"