如何检查Linux PyTorch安装

要检查在 Linux 上是否已成功安装 PyTorch，可以按照以下步骤进行：

1. 打开终端：首先，打开你的 Linux 终端。

2. 启动 Python 或 Python3 解释器：

   • 输入 python 或 python3 并按回车键，进入 Python 交互环境。具体使用哪个命令取决于你的系统配置。

3. 导入 PyTorch：

   • 在 Python 提示符下，尝试导入 PyTorch 库，输入以下命令：

     import torch 

4. 检查 PyTorch 版本：

   • 你可以打印 PyTorch 的版本号，以确保它已正确安装：

     print(torch.__version__) 

5. 验证 GPU 支持（可选）：

   • 如果你的系统有 NVIDIA GPU 并且已经安装了 CUDA，可以检查 PyTorch 是否能够检测到 GPU：

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

   • 如果输出是 cuda，则表示 PyTorch 可以使用 GPU。你还可以进一步检查可用的 GPU 数量：

     print(torch.cuda.device_count()) 

6. 测试简单的张量操作：

   • 为了进一步确认 PyTorch 是否正常工作，可以尝试创建一个简单的张量并进行一些基本操作：

     x = torch.rand(2, 3) y = torch.rand(2, 3) z = x + y print(z) 

如果在以上步骤中没有出现错误消息，并且版本号和设备信息显示正确，那么 PyTorch 应该已经在你的 Linux 系统上成功安装。如果在导入或运行过程中遇到错误，请检查你的安装步骤是否正确，或者查看 PyTorch 官方网站获取更多支持和安装指南。