如何检查Linux PyTorch安装是否成功

要确认 linux 系统中 pytorch 是否安装成功，可以参考以下操作流程：

1. 打开终端： 启动你的 Linux 终端程序。

2. 进入 python 环境： 根据系统设置，输入 python 或 python3 命令来启动 Python 交互界面。

    python

   或者使用

    python3

3. 尝试导入 PyTorch 模块： 在 Python 中试着加载 torch 和 torchvision 模块，以验证其可用性。

    import torch  import torchvision

4. 查看当前安装的 PyTorch 版本： 输出 PyTorch 的版本号，用于确认安装状态。

    print(torch.__version__)

5. 检测 GPU 驱动支持情况（如适用）： 如果你拥有 NVIDIA 显卡并已配置 CUDA 环境，可以通过下面的代码判断 PyTorch 能否识别到显卡设备。

    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")  print(device)

   若输出为 cuda，则表示 PyTorch 成功识别到了 GPU。

6. 执行基础张量运算测试： 最后，运行一个简单的张量创建和打印操作，进一步验证功能是否正常。

    x = torch.rand(2, 3)  print(x)

   正常情况下会显示一个 2 行 3 列的随机数值矩阵。

只要在完成以上所有步骤时未出现异常，并且能够正确获取版本与设备信息，就说明 PyTorch 已经顺利安装并可在你的 Linux 系统上使用。