import torch
print(torch.version.cuda)
查看cuda是否安装成功和cuda版本号
最新推荐文章于 2024-07-28 22:38:05 发布
import torch
print(torch.version.cuda)