作为深度学习领域最主流的框架之一,PyTorch的安装一直是开发者入门的第一个挑战。特别是在不同操作系统环境下,依赖项管理和硬件加速配置的差异常常让人头疼。本文将基于2025年最新稳定版本,详解Windows、Linux和macOS三大平台下的完整安装方案,涵盖从基础CPU版本到CUDA加速的全套配置流程。
实测环境:PyTorch 2.4 LTS版本(2025年4月发布),各系统均通过conda和pip两种方式验证
在开始安装前,需要根据目标平台做好以下准备:
Windows系统要求:
Linux系统要求:
macOS系统要求:
Conda优势场景:
Pip优势场景:
关键区别:Conda会处理非Python依赖(如CUDA运行时),而pip仅管理Python包
CPU版本基准测试:
bash复制# Conda方案
conda create -n pytorch_env python=3.10
conda activate pytorch_env
conda install pytorch torchvision torchaudio cpuonly -c pytorch
# Pip方案
python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
验证安装成功的标准操作:
python复制import torch
print(torch.__version__) # 应显示2.4.x
print(torch.rand(5,3)) # 生成随机矩阵测试基础功能
NVIDIA驱动选择:
CUDA Toolkit安装技巧:
常见坑点:系统PATH变量冲突,建议在安装后手动检查环境变量优先级
AMD显卡用户需采用ROCm方案:
bash复制# Ubuntu专用仓库
wget -q -O - https://repo.radeon.com/rocm/rocm.gpg.key | sudo apt-key add -
echo 'deb [arch=amd64] https://repo.radeon.com/rocm/apt/6.0 ubuntu main' | sudo tee /etc/apt/sources.list.d/rocm.list
# 安装核心组件
sudo apt update
sudo apt install rocm-llvm rocm-opencl-runtime
验证ROCm检测:
python复制torch.cuda.is_available() # 应返回True
torch.cuda.get_device_name(0) # 显示AMD显卡型号
Apple Silicon芯片专用配置:
bash复制conda install -c apple pytorch-metal
性能优化建议:
python复制device = torch.device("mps")
x = torch.randn(10000, device=device) # 显式指定设备
典型报错:
code复制ImportError: libcudart.so.12.4: cannot open shared object file
处理步骤:
bash复制nvcc --version # 查看编译器版本
conda list cudatoolkit # 查看conda环境版本
bash复制conda install cudatoolkit=12.4 -c nvidia
Windows特有方案:
python复制torch.set_num_threads(4) # 限制CPU并行度
Linux优化建议:
bash复制# 禁用透明大页
echo never > /sys/kernel/mm/transparent_hugepage/enabled
内核关联技巧:
bash复制python -m ipykernel install --user --name pytorch_env --display-name "PyTorch 2.4"
GPU监控插件:
python复制# 在notebook单元格中添加
!nvidia-smi # 实时显存监控
%load_ext torch.utils.tensorboard
官方镜像选择建议:
pytorch/pytorch:2.4.0-cuda12.4-cudnn8-runtimepytorch/pytorch:2.4.0-cuda12.4-cudnn8-devel多阶段构建示例:
dockerfile复制FROM nvidia/cuda:12.4-base as builder
RUN pip install torch==2.4.0 --extra-index-url https://download.pytorch.org/whl/cu124
FROM ubuntu:22.04
COPY --from=builder /usr/local/lib/python3.10/site-packages /usr/local/lib/python3.10/site-packages
从2.3.x升级到2.4.x的建议步骤:
bash复制conda env export > environment.yml
pip freeze > requirements.txt
bash复制pip install torch==2.4.0 --upgrade --dry-run
python复制# 特别注意已弃用方法
with torch.warnings.catch_warnings():
torch.warnings.filterwarnings("error", category=DeprecationWarning)
PyTorch 2.4 LTS的生命周期:
企业级维护建议: