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