检查CUDA是否可用的标准代码

最新推荐文章于 2025-06-10 19:30:00 发布

Obolicaca

最新推荐文章于 2025-06-10 19:30:00 发布

阅读量5.9k

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch 文章标签： python pytorch

本文链接：https://blog.csdn.net/Obolicaca/article/details/123958964

pytorch 专栏收录该内容

10 篇文章

订阅专栏

这段代码检查CUDA是否可用，如果可用则设置默认数据类型为CUDA浮点数，并显示设备名称。若不可用，设备将指向CPU，确保在无GPU环境下也能运行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

if torch.cuda.is_available(): 
    torch.set_default_tensor_type(torch. cuda. FloatTensor)
    print ("using cuda:",torch.cuda.get_device_name(0))
    pass 
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
device

检查的语句是torch.cuda.is_available()。如果设备可用，我们将默认的数据类型设为torch.cuda.FloatTensor。然后，我们打印出CUDA找到的设备名。这段标准代码同时设置一个PyTorch设备，在没有默认使用GPU的情况下，允许我们通过代码将数据传入GPU。如果CUDA不可用，则设备会指向CPU。