ARTICLE DETAIL

建站实战干货

来自一线的建站与推广经验沉淀,每一条都经过真实交付验证。

如何测试python 版本与 torch 、 torchvision 版本是否对应?

2026/9/29 4:51:27 拓冰建站 浏览量
如何测试python 版本与 torch 、 torchvision 版本是否对应?

python 版本与 torch 、 torchvision 版本的对应关系如下图所示:

 打开 anaconda powershell prompt,输入如下命令:

>python>>>import torch>>>c = torch.ones((3,1))   //创建矩阵>>>c = c.cuda(0)              //将该矩阵挪到本机的 GPU 上

若如上命令行执行过程中均无报错则表明本机上的 torch 与 torchvision是对应的。