检查torch是否是gpu版本

发布时间 2023-09-10 21:09:12作者: tommickey

检查torch是否是gpu版本

1. 查看PyTorch版本:

打开Python交互式环境,导入torch包,使用命令torch.__version__查看PyTorch版本,如果版本名称中包含“cuda”,则表示是GPU版本。

例如,如果版本名称为“1.7.0+cu101”,则是支持CUDA 10.1的GPU版本。

2. 查看torch.cuda:

在Python交互式环境中,导入torch包后,使用命令torch.cuda.is_available()检查CUDA是否可用。如果返回值为True,则表示是GPU版本。

3. 查看GPU设备列表:

在Python交互式环境中,导入torch包后,使用命令torch.cuda.device_count()检查当前系统中可用的GPU设备数量,如果返回值大于0,则表示是GPU版本。可以使用torch.cuda.get_device_name()命令查看每个设备的名称。

例如,如果返回值为1,并且使用torch.cuda.get_device_name(0)命令返回GPU设备的名称,则说明是GPU版本。