温馨提示×

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

小樊
63
2025-06-13 09:10:56
栏目: 智能运维

要检查在 Linux 上安装的 PyTorch 是否成功,可以按照以下步骤进行:

  1. 打开终端: 打开你的 Linux 终端。

  2. 启动 Python 解释器: 输入 pythonpython3(取决于你的系统配置),然后按回车键进入 Python 交互环境。

    python
    

    或者

    python3
    
  3. 导入 PyTorch: 在 Python 解释器中,尝试导入 PyTorch 库。你可以检查 torchtorchvision 是否可以成功导入。

    import torch
    import torchvision
    
  4. 检查 PyTorch 版本: 打印 PyTorch 的版本信息,以确保它已正确安装。

    print(torch.__version__)
    
  5. 验证 GPU 支持(如果有 GPU): 如果你的机器上有 NVIDIA GPU 并且已经安装了 CUDA,可以检查 PyTorch 是否能够检测到 GPU。

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

    如果 PyTorch 安装正确并且有可用的 GPU,你应该会看到输出为 cuda

  6. 运行简单的 PyTorch 操作: 你可以运行一些简单的 PyTorch 操作来确保一切正常工作。例如,创建一个张量并进行基本操作:

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

    这应该会输出一个 2x3 的随机张量。

如果在执行上述步骤时没有出现错误,并且版本号和设备信息正确,那么 PyTorch 应该已经成功安装在你的 Linux 系统上了。

0