HuaPu在学:torch和torchvision版本对应关系


前言


一、torch和torchvision版本对应关系

错误分析:
安装pytorch或torchvision时,无法找到对应版本
cuda可以找到,但是无法转为.cuda()

以上两种或类似错误,一般由两个原因可供分析:
cuda版本不合适,重新安装cuda和cudnn
pytorch和torchvision版本没对应上

安装torch和torchvision时,两个包都有版本对应关系
在这里插入图片描述

二、导入离线包

【注意导入离线包前提是将对应版本的包下载完成】
在导入前,建议大家导入到虚拟环境里。

1.打开虚拟化环境

代码如下(示例):

cmd打开创建好的虚拟环境
(base环境下的转入目录方法)
例如转入d:

![在这里插入图片描述](https://img-blog.csdnimg.cn/8604ca1328194485b383815cd17dc261.png
(虚拟环境下)
在这里插入图片描述
在这里插入图片描述

转入自己下载包的目录,之后输入导入命令pip install 包名.whl
当提示Successfully installed ....就说明安装好了。

2.验证是否安装成功

新建python文件或者打开交互式界面,输入代码

import torch
torch.__version__
==> 2.0.0
torch.cuda.is_available()
==> True
  • 13
    点赞
  • 61
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值