- 博客(17)
- 收藏
- 关注
原创 检测深度学习环境是GPU还是CPU(代码)
print(f"当前设备是GPU,设备名称为: {torch.cuda.get_device_name(0)}")device = torch.device("cuda") # 创建一个表示GPU设备的对象。device = torch.device("cpu") # 创建一个表示CPU设备的对象。# 使用torch.cuda.is_available()方法来判断是否有可用的GPU。print(f"当前使用的设备为: {device}")print("当前设备是CPU")
2025-02-27 18:45:20
270
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人