解决【cuda突然连接不上】+【cuda和torch版本不匹配】的方法

问题一、pytorch连不上cuda

在使用gpu+pytorch训练完模型的第二天,cuda就突然连不上了。

 print(torch.cuda.is_available())

一直显示False

 print(torch.__version__)

发现torch的version由原来的gpu版本突然变成了cpu版本。

没搞懂为什么突然版本变了,在网上搜了搜,可能是我下载什么包的时候冲突了,或者哪个指令把torch更新了一下。具体的我也说不准,只能暂时归咎于以上可能的原因了。

急匆匆把现有的torch、torchvision、torchaudio卸载了,然后去官网找gpu版本的下载命令,发现下载下来的还是cpu版本。【坑】

这是我年初时最开始配置环境时就踩过的坑,明明官网说是gpu版本,但货不对版。我也不懂为什么。
这也是我特此写博客记录下来的原因,因为年初的时候就因为这个浪费了很多时间,现在又踩了同样的坑,又浪费了半天。以后绝对不会再踩了!

最后使用以下指令,成功解决连接cuda问题,但是又出现了新的问题

pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html

运行结果:
在这里插入图片描述

问题二、pytorch和cuda版本不匹配

新的问题:【这个坑我年初的时候好像也踩过……】
pytorch版本【11.8】和cuda版本【12.1】不匹配
在这里插入图片描述

但是当时下载的时候,明明说的就是12.1cuda对应的pytorch下载指令,包括官网中也有同样的报错信息出现【如下图】。

在这里插入图片描述

根据ptrblck说的,要么把cuda降级到11.8,要么升级pytorch到12.1,所以我又把原来的torch删除了,重新下载。至于为什么会不匹配,我没细看。官网链接在此【官网】,有兴趣的可以看一下。

执行下列语句,问题解决。

pip uninstall torch
pip install torch==2.1.2+cu121 -f https://download.pytorch.org/whl/torch_stable.html

现在看来这个博客也没啥特别有用的。针对遇到的两个问题的解决方法,不过就都是简单的安装语句罢了。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值