是否使用gpu的判断方法_怎么查看代码需要不需要gpu

作者：神奇cpp | 2024-08-06 20:41:10

踩

怎么查看代码需要不需要gpu

运行以下代码：

# import torch
# print("是否可用：", torch.cuda.is_available())        # 查看GPU是否可用
# print("GPU数量：", torch.cuda.device_count())        # 查看GPU数量
# print("torch方法查看CUDA版本：", torch.version.cuda)  # torch方法查看CUDA版本
# print("GPU索引号：", torch.cuda.current_device())    # 查看GPU索引号
# print("GPU名称：", torch.cuda.get_device_name(1))    # 根据索引号得到GPU名称

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/神奇cpp/article/detail/939185