要检查在 Linux 上安装的 PyTorch 是否成功,可以按照以下步骤进行:
打开终端: 打开你的 Linux 终端。
启动 Python 解释器:
输入 python 或 python3(取决于你的系统配置),然后按回车键进入 Python 交互环境。
python
或者
python3
导入 PyTorch:
在 Python 解释器中,尝试导入 PyTorch 库。你可以检查 torch 和 torchvision 是否可以成功导入。
import torch
import torchvision
检查 PyTorch 版本: 打印 PyTorch 的版本信息,以确保它已正确安装。
print(torch.__version__)
验证 GPU 支持(如果有 GPU): 如果你的机器上有 NVIDIA GPU 并且已经安装了 CUDA,可以检查 PyTorch 是否能够检测到 GPU。
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)
如果 PyTorch 安装正确并且有可用的 GPU,你应该会看到输出为 cuda。
运行简单的 PyTorch 操作: 你可以运行一些简单的 PyTorch 操作来确保一切正常工作。例如,创建一个张量并进行基本操作:
x = torch.rand(2, 3)
print(x)
这应该会输出一个 2x3 的随机张量。
如果在执行上述步骤时没有出现错误,并且版本号和设备信息正确,那么 PyTorch 应该已经成功安装在你的 Linux 系统上了。