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