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

2023-05-25 08:19:14 +08:00
 gpt5
但是 nvdia-smi 与 nvcc --version 都可以看到 cuda 版本。

咋解决啊??
2313 次点击
所在节点    问与答
11 条回复
Weixiao0725
2023-05-25 08:31:16 +08:00
用 docker 阿
1311317
2023-05-25 09:01:19 +08:00
版本不对应?不兼容?
JustSong
2023-05-25 09:05:38 +08:00
核对 Python 版本、CUDA 版本、cuDNN 版本以及 Tensorflow 版本是否兼容
superchijinpeng
2023-05-25 09:08:37 +08:00
确定装的是 tensorflow-gpu 吗
gpt5
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
2023-05-25 09:22:24 +08:00
tensorflow 太难装了 试过两次都没成功 最新版驱动应该是不行的 最后只能 docker 了
JustSong
2023-05-25 09:31:32 +08:00
@gpt5 #5 别只看驱动版本啊
julyclyde
2023-05-25 11:36:47 +08:00
@Weixiao0725 你难道不觉得这事恰恰是因为 docker 引起的吗?
WonderfulRush
2023-05-25 13:36:40 +08:00
tansorflow 安装根据官网教程来,并且选取英文的安装教程(中文可能是老版本的),基本没问题。
JerryJerry
2023-05-25 14:28:33 +08:00
我也遇到过好多次这种事情了 tensorflow 真心不好装
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 里会显示找不到驱动。

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/942733

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX