笔者在本地电脑配置 GPU 加速的 pytorch 环境,步骤如下

  1. 安装 CUDA 12.4

  2. 安装 cuDNN(专门针对深度神经网络而设计基于GPU的加速库)

  3. 安装 python3.14

  4. 安装 pytorch 时出现问题:
    ERROR: Could not find a version that satisfies the requirement torch2.5.1 (from versions: none)
    ERROR: No matching distribution found for torch
    2.5.1

    在这里插入图片描述

    搜索到下列回答:
    Installation is broken: pip3 install torch torchvision torchaudio fails

    上述回答说 python3.13 存在这个 bug,可以安装 python3.12。于是笔者卸载 python3.14 重装 python3.12

    果然没问题了!

    在这里插入图片描述

Logo

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐