Handsome_zhou 发表于 2022-3-7 11:52:54

测试Pytorch能不能调用cuda加速

检测CUDA是否安装正确并能被Pytorch检测到：

torch.cuda.is_available()


测试Pytorch能不能调用cuda加速：

a = torch.Tensor()
a = a.cuda()
a

页: [1]
查看完整版本: 测试Pytorch能不能调用cuda加速