pytorch一直可以使用cuda进行训练,但是重启了一次电脑,系统(Ubuntu 20.04.3 LTS)提示有更新,所以做了一下更新,重启之后就识别不到cuda了。
import torch
flag = torch.cuda.is_available()
print(flag)
False
/home/suned/anaconda3/envs/pytorch/lib/python3.8/site-packages/torch/cuda/init.py:80: UserWarning: CUDA initialization: CUDA driver initialization failed, you might not have a CUDA gpu. (Triggered internally at /opt/conda/conda-bld/pytorch_1639180544224/work/c10/cuda/CUDAFunctions.cpp:112.)
return torch._C._cuda_getDeviceCount() > 0
重装了cuda,依然返回找不到。
nvid-smi返回结果
请问这个问题改如何解决,搞了一天半了,严重影响进度啊。
说几个思路吧:
1.这是个人gpu还是公用,公用的话利用占显存的方式你不一定能读到这块gpu
2.看看你的pytorch版本和cuda版本对应了嘛,不对应的话是用不了的
3.重装cuda的时候,cudnn有没有挪进去
4.还有一个可能就是你更新以后,环境变量配置发生了变化,看看环境变量配置对不对
5.还有就是很多新装的东西需要重启才能生效,有没有重启