PyTorch CPU版本安装(Ubuntu环境下)

551 阅读1分钟

本博客供刚接触人工智能的小白或忘记相关内容的开发者配置环境时参考。网络上相关的文章浩如烟海,笔者仅希望这系列文章能为以后的自己或和我使用同样开发环境的开发者提供帮助。

注:笔者的操作系统环境是ubuntu22.04

1. PyTorch官网

pytorch官网:pytorch.org

对于新手而言,为了能立刻体验pytorch,可以直接安装pytoch的cpu版本 image-20230330225350854.png

2. 在线安装

2.1 在线安装cpu版本

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

image-20230330225132140.png

2.2 成功安装界面

image-20230330225753485.png

2.3 测试是否安装成功

进入python环境,输入下面的代码

import torch
print(torch.__version__)

image-20230330230003290.png

成功显示了pytorch的版本号,成功安装

至此您已成功安装了PyTorch的CPU版本