首页 > 编程知识 正文

pytorch查看版本命令,如何查看cuda版本

时间:2023-05-06 18:37:21 阅读:191774 作者:3574

由于pytorch的whl 安装包名字都一样,所以我们很难区分到底是基于cuda 的哪个版本。

有一条指令可以查看

import torch

print(torch.version.cuda)

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。