能够检验是否成功安装pytorch, cuda, cudnn,并能输出GPU型号
import torch
a = torch.cuda.is_available()
print(a)
ngpu= 1
device = torch.device("cuda:0" if (torch.cuda.is_available() <
本文介绍如何检查PyTorch、CUDA和CUDNN的安装情况,以及获取并显示GPU型号。通过实例代码演示了如何判断设备类型并生成随机张量。
能够检验是否成功安装pytorch, cuda, cudnn,并能输出GPU型号
import torch
a = torch.cuda.is_available()
print(a)
ngpu= 1
device = torch.device("cuda:0" if (torch.cuda.is_available() <
4905
10万+
3万+
3399
1089
7603

被折叠的 条评论
为什么被折叠?
