SunnyEdward 2022-01-28 09:48 采纳率: 75%
浏览 340
已结题

关于pytorch无法识别cuda的问题

问题遇到的现象和发生背景

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返回结果

img


nvcc --version返回结果

img

我想要达到的结果

请问这个问题改如何解决,搞了一天半了,严重影响进度啊。

  • 写回答

1条回答 默认 最新

  • 我是周大欣 2022-01-28 10:52
    关注

    说几个思路吧:
    1.这是个人gpu还是公用,公用的话利用占显存的方式你不一定能读到这块gpu
    2.看看你的pytorch版本和cuda版本对应了嘛,不对应的话是用不了的
    3.重装cuda的时候,cudnn有没有挪进去
    4.还有一个可能就是你更新以后,环境变量配置发生了变化,看看环境变量配置对不对
    5.还有就是很多新装的东西需要重启才能生效,有没有重启

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

问题事件

  • 系统已结题 2月6日
  • 已采纳回答 1月29日
  • 创建了问题 1月28日

悬赏问题

  • ¥20 请问这种量表怎么用spss量化分析(作为中介模型的因变量
  • ¥55 AD844 howland电流源如何驱动大额负载
  • ¥15 C++ /QT 内存权限的判断函数列举
  • ¥15 深度学习GFnet理解问题
  • ¥15 单细胞小提琴堆叠图代码
  • ¥50 升级strust2版本到2.3.15.1后使用ognl3.0.6.jar windows环境中没有问题,但部署到linux环境报错
  • ¥15 vue页面,node封装接口
  • ¥15 求TMS320F280039C工程模板!
  • ¥15 delphi+fastreport实现分组补空打印问题
  • ¥15 使用python把两台mysql数据库服务器数据导出和导入