Handsome_zhou 发表于 2023-6-26 15:52:47

检查cuda是否安装成功

本帖最后由 Handsome_zhou 于 2023-6-26 15:55 编辑

import torch

torch.cuda.is_available()

a = torch.Tensor()
a = a.cuda()
a
页: [1]
查看完整版本: 检查cuda是否安装成功