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

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

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

复制代码
>python

>>>import torch

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

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

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

相关推荐
陈天伟教授11 小时前
人工智能训练师认证教程(2)Python os入门教程
前端·数据库·python
2301_7644413311 小时前
Aella Science Dataset Explorer 部署教程笔记
笔记·python·全文检索
爱笑的眼睛1111 小时前
GraphQL:从数据查询到应用架构的范式演进
java·人工智能·python·ai
江上鹤.14811 小时前
Day40 复习日
人工智能·深度学习·机器学习
BoBoZz1911 小时前
ExtractSelection 选择和提取数据集中的特定点,以及如何反转该选择
python·vtk·图形渲染·图形处理
liwulin050611 小时前
【PYTHON-YOLOV8N】如何自定义数据集
开发语言·python·yolo
行如流水12 小时前
BLIP和BLIP2解析
深度学习
木头左12 小时前
LSTM量化交易策略中时间序列预测的关键输入参数分析与Python实现
人工智能·python·lstm
电子硬件笔记12 小时前
Python语言编程导论第七章 数据结构
开发语言·数据结构·python
fegggye12 小时前
PyO3 Class 详解 - 在 Python 中使用 Rust 类
pytorch·rust