【pytorch】查看torch的gpu是否可用

2022-10-23 08:55:37

cuda是否可用

torch.cuda.is_available()

返回gpu数量

torch.cuda.device_count()

返回gpu名字，设备索引默认从0开始

torch.cuda.get_device_name(0)

返回当前设备索引

torch.cuda.current_device()
  • 作者：Jack_Kuo
  • 原文链接：https://jack-kuo.blog.csdn.net/article/details/124117989
    更新时间：2022-10-23 08:55:37