
要确认 Linux 系统中 PyTorch 是否安装成功,可以参考以下操作流程:
打开终端: 启动你的 Linux 终端程序。
-
进入 Python 环境: 根据系统设置,输入 python 或 python3 命令来启动 Python 交互界面。
<code> python</code>
或者使用
<code> python3</code>
-
尝试导入 PyTorch 模块: 在 Python 中试着加载 torch 和 torchvision 模块,以验证其可用性。
<code> import torch import torchvision</code>
-
查看当前安装的 PyTorch 版本: 输出 PyTorch 的版本号,用于确认安装状态。
<code> print(torch.__version__)</code>
-
检测 GPU 驱动支持情况(如适用): 如果你拥有 NVIDIA 显卡并已配置 CUDA 环境,可以通过下面的代码判断 PyTorch 能否识别到显卡设备。
<code> device = torch.device("cuda" if torch.cuda.is_available() else "cpu") print(device)</code>若输出为 cuda,则表示 PyTorch 成功识别到了 GPU。
-
执行基础张量运算测试: 最后,运行一个简单的张量创建和打印操作,进一步验证功能是否正常。
<code> x = torch.rand(2, 3) print(x)</code>
正常情况下会显示一个 2 行 3 列的随机数值矩阵。
只要在完成以上所有步骤时未出现异常,并且能够正确获取版本与设备信息,就说明 PyTorch 已经顺利安装并可在你的 Linux 系统上使用。










