pytorch安装之后测试cuda是否安装成功命令

lightHos

已于 2024-10-11 21:37:19 修改

阅读量2.1k

点赞数 2

分类专栏：笔记文章标签： pytorch python 深度学习

于 2022-07-07 14:39:32 首次发布

本文链接：https://blog.csdn.net/qq_45601448/article/details/125658827

版权

笔记专栏收录该内容

4 篇文章 3 订阅

订阅专栏

当我们安装好cuda和pytorch之后，需要测试安装的环境是否可用，所以，可以用下面代码测试

def print_torch_test():
    print("pyTorch版本:",torch.__version__)  # 输出pyTorch版本
    print("CUDA版本:",torch.version.cuda)  # 输出CUDA版本
    print("cuDNN的版本:",torch.backends.cudnn.version())  # 输出cuDNN的版本
    print("显卡:",torch.cuda.get_device_name(0))  # 输出显卡版本
    print(torch.cuda.is_available())  # 测试GPU是否生效