要检查在 Linux 上是否已成功安装 PyTorch,可以按照以下步骤进行:
-
打开终端:首先,打开你的 Linux 终端。
-
启动 Python 或 Python3 解释器:
- 输入
python
或python3
并按回车键,进入 Python 交互环境。具体使用哪个命令取决于你的系统配置。
- 输入
-
导入 PyTorch:
- 在 Python 提示符下,尝试导入 PyTorch 库,输入以下命令:
import torch
- 在 Python 提示符下,尝试导入 PyTorch 库,输入以下命令:
-
检查 PyTorch 版本:
- 你可以打印 PyTorch 的版本号,以确保它已正确安装:
print(torch.__version__)
- 你可以打印 PyTorch 的版本号,以确保它已正确安装:
-
验证 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())
- 如果你的系统有 NVIDIA GPU 并且已经安装了 CUDA,可以检查 PyTorch 是否能够检测到 GPU:
-
测试简单的张量操作:
- 为了进一步确认 PyTorch 是否正常工作,可以尝试创建一个简单的张量并进行一些基本操作:
x = torch.rand(2, 3) y = torch.rand(2, 3) z = x + y print(z)
- 为了进一步确认 PyTorch 是否正常工作,可以尝试创建一个简单的张量并进行一些基本操作:
如果在以上步骤中没有出现错误消息,并且版本号和设备信息显示正确,那么 PyTorch 应该已经在你的 Linux 系统上成功安装。如果在导入或运行过程中遇到错误,请检查你的安装步骤是否正确,或者查看 PyTorch 官方网站获取更多支持和安装指南。