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

衹染青春2024-01-24 9:20

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

打开 anaconda powershell prompt，输入如下命令：

复制代码

>python

>>>import torch

>>>c = torch.ones((3,1))   //创建矩阵

>>>c = c.cuda(0)              //将该矩阵挪到本机的 GPU 上

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