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

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

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

复制代码
>python

>>>import torch

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

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

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

相关推荐
MgArcher1 分钟前
Python 入门核心考点:数据类型与变量全解
python
m0_662577975 分钟前
自动化机器学习(AutoML)库TPOT使用指南
jvm·数据库·python
q_354888515310 分钟前
计算机毕业设计源码:Python动漫智能推荐与可视化分析系统 Django框架 协同过滤推荐算法 可视化 数据分析 大数据 大模型(建议收藏)✅
python·scrapy·数据分析·django·课程设计·旅游·推荐算法
与虾牵手10 分钟前
写了 8 年 Python 转 Rust,我被所有权机制折磨了整整一周
python
ghie909016 分钟前
基于HHT与神经网络结合的故障诊断系统
人工智能·深度学习·神经网络
sqyno1sky18 分钟前
机器学习模型部署:将模型转化为Web API
jvm·数据库·python
何政@27 分钟前
通过python 快速完成ai 构建
人工智能·python·ai·大模型·love l
万粉变现经纪人32 分钟前
如何解决 pip install shapely 报错 GEOS C 库未找到 问题
c语言·开发语言·python·pycharm·bug·pandas·pip
dapeng287034 分钟前
机器学习与人工智能
jvm·数据库·python
YOLO视觉与编程1 小时前
一文读懂-yolo26如何预测识别图片|视频|摄像头|文件夹检测适用v8v11
人工智能·深度学习·yolo·目标检测·机器学习