import torch
device = 'cuda' if torch.cuda.is_available() else 'cpu'
【Pytorch】GPU不可用,则使用CPU
最新推荐文章于 2024-10-16 11:51:36 发布
本文介绍了如何在Python中使用PyTorch库检查设备可用性,通过`torch.cuda.is_available()`函数判断是否支持CUDA,从而决定在GPU(如cuda)或CPU(如cpu)上运行代码。
摘要由CSDN通过智能技术生成