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

关于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 LD衰减图用R语言对其可视化
  • ¥15 Mermaid语法生成的svg在Axure无法编辑
  • ¥15 Windchill二次开发
  • ¥15 怎么把tkinter和其他代码结合起来使功能可以完善
  • ¥15 爬取豆瓣电影相关处理
  • ¥15 手机淘宝抓清除消息接口
  • ¥15 C#无selenium
  • ¥15 用机器学习方法帮助保险公司预测哪些是欺诈行为
  • ¥15 数据爬取,python
  • ¥15 怎么看 cst中一个面的功率分布图,请说明详细步骤。类似下图