检查PyTorch版本及GPU是否安装成功检测

2023-12-16 21:51:12
import torch
print（torch.__version__)
print(torch.cuda.is_available())

前两行代码是输出版本

后面那一行，输出True 就代表安装成功，反之安装失败

文章来源:https://blog.csdn.net/weixin_72965172/article/details/134921242
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。