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

要确认 Linux 系统中 PyTorch 是否安装成功，可以参考以下操作流程：

1. 打开终端： 启动你的 Linux 终端程序。

2. 进入 Python 环境： 根据系统设置，输入 python 或 python3 命令来启动 Python 交互界面。

   <code> python</code>

   登录后复制

   或者使用

   <code> python3</code>

   登录后复制

3. 尝试导入 PyTorch 模块： 在 Python 中试着加载 torch 和 torchvision 模块，以验证其可用性。

   <code> import torch
    import torchvision</code>

   登录后复制

4. 查看当前安装的 PyTorch 版本： 输出 PyTorch 的版本号，用于确认安装状态。

   

   帮衣帮-AI服装设计

   AI服装设计神器，AI生成印花、虚拟试衣、面料替换

   106

   查看详情

   <code> print(torch.__version__)</code>

   登录后复制

5. 检测 GPU 驱动支持情况（如适用）： 如果你拥有 NVIDIA 显卡并已配置 CUDA 环境，可以通过下面的代码判断 PyTorch 能否识别到显卡设备。

   <code> device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
    print(device)</code>

   登录后复制

   若输出为 cuda，则表示 PyTorch 成功识别到了 GPU。

6. 执行基础张量运算测试： 最后，运行一个简单的张量创建和打印操作，进一步验证功能是否正常。

   <code> x = torch.rand(2, 3)
    print(x)</code>

   登录后复制

   正常情况下会显示一个 2 行 3 列的随机数值矩阵。

只要在完成以上所有步骤时未出现异常，并且能够正确获取版本与设备信息，就说明 PyTorch 已经顺利安装并可在你的 Linux 系统上使用。

以上就是如何检查Linux PyTorch安装是否成功的详细内容，更多请关注php中文网其它相关文章！