import torch
n_gpu = torch.cuda.device_count()
print(n_gpu)
检查自己电脑GPU个数
最新推荐文章于 2024-04-24 16:47:27 发布
import torch
n_gpu = torch.cuda.device_count()
print(n_gpu)