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

关于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日

悬赏问题

  • ¥15 PADS Logic 原理图
  • ¥15 PADS Logic 图标
  • ¥15 电脑和power bi环境都是英文如何将日期层次结构转换成英文
  • ¥20 气象站点数据求取中~
  • ¥15 如何获取APP内弹出的网址链接
  • ¥15 wifi 图标不见了 不知道怎么办 上不了网 变成小地球了