大家好,又见面了,我是你们的朋友全栈君。
由于pytorch的whl 安装包名字都一样,所以我们很难区分到底是基于cuda 的哪个版本。
有一条指令可以查看
import torch
print(torch.version.cuda)
发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/135661.html原文链接:https://javaforall.net
pytorch 查看cuda 版本由于pytorch的whl安装包名字都一样,所以我们很难区分到底是基于cuda的哪个版本。有一条指令可以查看importtorchprint(torch.version.cuda)
大家好,又见面了,我是你们的朋友全栈君。
由于pytorch的whl 安装包名字都一样,所以我们很难区分到底是基于cuda 的哪个版本。
有一条指令可以查看
import torch
print(torch.version.cuda)
发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/135661.html原文链接:https://javaforall.net