PyTorch安装教程(适合新手)

安装前准备

  1. 安装Python
    • 确认已安装 Python 3.6及以上版本(PyTorch不支持Python 2)。
    • 检查方法:命令行输入 python --version,若未安装,从Python官网下载安装,勾选“Add Python to PATH”

  2. 选择包管理工具
    推荐使用Anaconda(管理依赖更方便):
    下载 Anaconda,安装时勾选“Add to PATH”。


安装PyTorch

方法一:使用Anaconda(推荐)

  1. 打开Anaconda Prompt(Windows)或终端(Mac/Linux)。
  2. 创建虚拟环境(可选但推荐)
    避免与其他项目冲突,执行:
    conda create -n pytorch_env python=3.8  # 创建名为pytorch_env的环境
    conda activate pytorch_env              # 激活环境
    
  3. 复制安装命令
    • 访问 PyTorch官网,按如下图选择:
    ◦ Package:Conda 或 Pip(建议Conda)
    ◦ 无GPU电脑:选择 CPU
    ◦ 有NVIDIA GPU:选择CUDA版本(需提前安装NVIDIA驱动
    • 示例(CPU版本):
    conda install pytorch torchvision torchaudio cpuonly -c pytorch
    

方法二:使用Pip

  1. 在命令行输入官网提供的Pip命令,例如:
    pip install torch torchvision torchaudio
    

分平台注意事项

Windows:安装时可能需要Visual C++ Redistributable,从微软官网下载安装。
Mac/Linux:通常无需额外配置。


验证安装

  1. 打开Python环境,输入以下代码:
    import torch
    print(torch.__version__)          # 查看PyTorch版本
    print(torch.cuda.is_available())  # 输出True表示支持GPU
    
    • 预期结果示例:
    2.1.0
    False  # 若为True则GPU可用
    

常见问题

  1. 提示“conda/pip不是命令”
    • 检查Anaconda/Python是否安装成功,重启终端或计算机。

  2. 下载速度慢或失败
    更换国内镜像源
    Conda:

    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --set show_channel_urls yes
    

    Pip:

    pip install 包名 -i https://pypi.tuna.tsinghua.edu.cn/simple
    
  3. 导入torch时报错
    • 确认安装的PyTorch版本与Python版本匹配。
    • 尝试重新安装:conda uninstall pytorch → 重新执行安装命令。


学习资源推荐

官方教程PyTorch官方文档
新手入门:《动手学深度学习》(含PyTorch实现)
实践项目:从简单的线性回归、MNIST手写数字识别开始。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐