文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

服务器已经存在CUDA但无法使用GPU

2023-10-11 11:44

关注

记录一下我使用服务器期间遇到的问题以及解决方法(主要使用tensorflow)

先介绍一下我本次遇到的问题:

我在服务器上拥有一个独立的账号,我发现我的tensorflow无法调用GPU,先排查可能存在的问题

终端输入

nvcc -V

结果如下:

显示已经安装了11.8版本的CUDA

 但是在python文件中调用

import tensorflow as tfprint(tf.test.is_gpu_available())

结果显示为false

 在终端输入

echo $LD_LIBRARY_PATH

结果为空,猜测是没有安装cudnn,以下是解决过程

首先进入官网下载一个适配的cudnn版本,官网链接

因为我的CUDA版本是11.8,我选择v8.9.0 for CUDA 11.x的版本

下载后是一个.tar.xz的文件,直接上传到服务器中你的文件夹下,然后使用下面两条指令进行解压

xz -d cudnn-linux-x86_64-8.9.0.131_cuda11-archive.tar.xztar -xvf cudnn-linux-x86_64-8.9.0.131_cuda11-archive.tar

 解压得到一个文件夹,进入后包含下面三个

 然后我们进入你个人账户下的.bashrc文件,在其中添加这样一行内容(根据你自己的路径进行更改):

export LD_LIBRARY_PATH="/home/Users/qjw/cudnn-linux-x86_64-8.9.0.131_cuda11-archive/lib:$LD_LIBRARY_PATH"

之后保存.bashrc文件,然后在终端输入下面这个指令(我是在.bashrc文件这一层输入的,注意终端路径

source ./.bashrc

然后再次运行python代码:

import tensorflow as tfprint(tf.test.is_gpu_available())

 结果如下:

 输出True,问题解决

希望可以帮到你

来源地址:https://blog.csdn.net/qq_48494339/article/details/131518249

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-服务器
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯