查看GPU的序列或者型号

最新推荐文章于 2025-02-14 14:51:45 发布

Cen.c

最新推荐文章于 2025-02-14 14:51:45 发布

阅读量287

点赞数 4

文章标签： python 深度学习开发语言

本文链接：https://blog.youkuaiyun.com/Cen_cen_/article/details/142662803

版权

>>> import torch         
>
cuda是否可用            
>>>print(torch.cuda.is_available())
True

返回gpu数量；
>>> print(torch.cuda.device_count()) 
1

返回当前设备索引；
>>> print(torch.cuda.current_device())
0

返回gpu名字，设备索引默认从0开始；
>>> print(torch.cuda.get_device_name(0))
NVIDIA GeForce RTX 2080 Super