Pytorch和torchvision版本查询

在安装完pytorch和torchvision后,可能需要查询对应的版本。方法如下:

1.在安装的环境下输入python,然后输入以下代码查询pytorch的版本:

import torch
print(torch.__version__)

2.输入以下代码查询torchvision的版本:

import torchvision
print(torchvision.__version__)

示例如下:
在这里插入图片描述
下面放一个python、pytorch、torchvison和cuda的版本对应
在这里插入图片描述

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐