python 版本与 torch 、 torchvision 版本的对应关系如下图所示:
打开 anaconda powershell prompt,输入如下命令:文章来源:https://www.toymoban.com/news/detail-820283.html
>python
>>>import torch
>>>c = torch.ones((3,1)) //创建矩阵
>>>c = c.cuda(0) //将该矩阵挪到本机的 GPU 上
若如上命令行执行过程中均无报错则表明本机上的 torch 与 torchvision是对应的。文章来源地址https://www.toymoban.com/news/detail-820283.html
到了这里,关于如何测试python 版本与 torch 、 torchvision 版本是否对应?的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!