要确认 Linux 系统中 PyTorch 是否安装成功,可以参考以下操作流程:
打开终端: 启动你的 Linux 终端程序。
进入 Python 环境: 根据系统设置,输入 python 或 python3 命令来启动 Python 交互界面。
python
或者使用
python3
尝试导入 PyTorch 模块: 在 Python 中试着加载 torch 和 torchvision 模块,以验证其可用性。
import torch import torchvision
查看当前安装的 PyTorch 版本: 输出 PyTorch 的版本号,用于确认安装状态。
print(torch.__version__)
检测 GPU 驱动支持情况(如适用): 如果你拥有 NVIDIA 显卡并已配置 CUDA 环境,可以通过下面的代码判断 PyTorch 能否识别到显卡设备。
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") print(device)
若输出为 cuda,则表示 PyTorch 成功识别到了 GPU。
执行基础张量运算测试: 最后,运行一个简单的张量创建和打印操作,进一步验证功能是否正常。
x = torch.rand(2, 3) print(x)
正常情况下会显示一个 2 行 3 列的随机数值矩阵。
只要在完成以上所有步骤时未出现异常,并且能够正确获取版本与设备信息,就说明 PyTorch 已经顺利安装并可在你的 Linux 系统上使用。
以上就是如何检查Linux PyTorch安装是否成功的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号