import torch
print（torch.__version__)
print(torch.cuda.is_available())
前两行代码是输出版本
后面那一行，输出True 就代表安装成功，反之安装失败
检查PyTorch版本及GPU是否安装成功检测
猜你喜欢
转载自blog.csdn.net/weixin_72965172/article/details/134921242
今日推荐
周排行