V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
V2EX 提问指南
gpt5
V2EX  ›  问与答

装 tensorflow 提示找不到 cuda: Could not find cuda drivers on your machine, GPU will not be used.

  •  
  •   gpt5 · 2023-05-25 08:19:14 +08:00 · 2276 次点击
    这是一个创建于 550 天前的主题,其中的信息可能已经有所发展或是发生改变。
    但是 nvdia-smi 与 nvcc --version 都可以看到 cuda 版本。

    咋解决啊??
    11 条回复    2023-05-25 21:58:36 +08:00
    Weixiao0725
        1
    Weixiao0725  
       2023-05-25 08:31:16 +08:00
    用 docker 阿
    1311317
        2
    1311317  
       2023-05-25 09:01:19 +08:00 via Android
    版本不对应?不兼容?
    JustSong
        3
    JustSong  
       2023-05-25 09:05:38 +08:00 via Android
    核对 Python 版本、CUDA 版本、cuDNN 版本以及 Tensorflow 版本是否兼容
    superchijinpeng
        4
    superchijinpeng  
       2023-05-25 09:08:37 +08:00
    确定装的是 tensorflow-gpu 吗
    gpt5
        5
    gpt5  
    OP
       2023-05-25 09:20:33 +08:00
    @1311317 @JustSong  
    我也感觉是版本依存问题,但是 driver 是 530 ,cuda 文件名屁股上也带着 530 ,应该是对应的啊。

    @superchijinpeng
    我记得以前是可以装 tensorflow-gpu 来解决,但是现在提示 tenorflow-gpu 已经被 remove 了,让直接装 tensorflow 。

    @Weixiao0725 我用 pycharm 远程开发,没进过远程的 docker 。。实在解决不了的话就用 docker 了。
    ltkun
        6
    ltkun  
       2023-05-25 09:22:24 +08:00 via Android
    tensorflow 太难装了 试过两次都没成功 最新版驱动应该是不行的 最后只能 docker 了
    JustSong
        7
    JustSong  
       2023-05-25 09:31:32 +08:00 via Android
    @gpt5 #5 别只看驱动版本啊
    julyclyde
        8
    julyclyde  
       2023-05-25 11:36:47 +08:00
    @Weixiao0725 你难道不觉得这事恰恰是因为 docker 引起的吗?
    WonderfulRush
        9
    WonderfulRush  
       2023-05-25 13:36:40 +08:00   ❤️ 1
    tansorflow 安装根据官网教程来,并且选取英文的安装教程(中文可能是老版本的),基本没问题。
    JerryJerry
        10
    JerryJerry  
       2023-05-25 14:28:33 +08:00 via iPhone
    我也遇到过好多次这种事情了 tensorflow 真心不好装
    Weixiao0725
        11
    Weixiao0725  
       2023-05-25 21:58:36 +08:00
    @julyclyde

    https://stackoverflow.com/questions/75614728/cuda-12-tf-nightly-2-12-could-not-find-cuda-drivers-on-your-machine-gpu-will

    大概看了下,好像 docker 确实解决不了。是因为装的显卡驱动太新了,tf 还不支持导致的,所以 nvidia-smi 可以正常看到驱动,但是 tf 里会显示找不到驱动。
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   4731 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 24ms · UTC 10:06 · PVG 18:06 · LAX 02:06 · JFK 05:06
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.