检查PyTorch版本及GPU是否安装成功检测

作者:宇宙中心我曹县2024.01.07 17:26浏览量:3

简介:本文将指导您如何检查PyTorch的版本以及GPU是否已正确安装和运行。

千帆应用开发平台“智能体Pro”全新上线 限时免费体验

面向慢思考场景,支持低代码配置的方式创建“智能体Pro”应用

立即体验

首先,我们可以通过Python命令行检查PyTorch的版本。在Python环境下,键入以下命令:

  1. import torch
  2. print(torch.__version__)

如果成功导入了PyTorch库,将显示已安装的PyTorch版本。
接下来,我们需要检查GPU是否已正确安装并运行。在Python环境下,键入以下命令:

  1. import torch
  2. print(torch.cuda.is_available())

如果返回值为True,表示GPU已正确安装并运行。如果返回值为False,表示GPU可能未正确安装或未运行。
为了进一步检查GPU的详细信息,我们可以使用以下命令:
```python
import torch
print(torch.cuda.device_count()) # 查看GPU数量
print(torch.cuda.current_device()) # 查看当前使用的GPU设备编号
print(torch.cuda.get_device_name(0)) # 查看第一个GPU设备的名称

article bottom image
图片