简介:本文将指导您如何检查PyTorch的版本以及GPU是否已正确安装和运行。
面向慢思考场景,支持低代码配置的方式创建“智能体Pro”应用
首先,我们可以通过Python命令行检查PyTorch的版本。在Python环境下,键入以下命令:
import torch
print(torch.__version__)
如果成功导入了PyTorch库,将显示已安装的PyTorch版本。
接下来,我们需要检查GPU是否已正确安装并运行。在Python环境下,键入以下命令:
import torch
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设备的名称