python tensorflow / CUDA:未检测到GPU

zvokhttg  于 2023-01-11  发布在  Python
关注(0)|答案(1)|浏览(414)

我有两台配备NVIDIA GeForce RTX 3060 GPU的Windows 11笔记本电脑,我想在上面运行Tensorflow。
如果这很重要的话,这两款笔记本电脑都是联想Legion 5笔记本电脑,“GPU工作模式”设置为“混合自动模式”。
第一台笔记本电脑具有以下设置:

Python     3.10.7
Tensorflow 2.9.1
CUDA       11.2.0
cuDNN      8.1.1

CPU        AMD Ryzen 7 6800H
GPU0       NVIDIA GeForce RTX 3060
GPU1       AMD Radeon Graphics

第二台笔记本电脑具有以下设置:

Python     3.10.9 Virtual Environment
Tensorflow 2.11.0
CUDA       11.2.2
cuDNN      8.1.1

CPU        Intel Core i7 12th Gen 12700H
GPU0       Intel Iris Xe
GPU1       NVIDIA GeForce RTX 3060

CUDA和cuDNN按照此视频安装:https://www.youtube.com/watch?v=hHWkvEcDBO0(除了conda部分)。
在第一台笔记本电脑上,一切正常,但在第二台笔记本电脑上,当执行tf.config.list_physical_devices('GPU')时,我得到一个空列表。
我试过将CUDA_VISIBLE_DEVICES变量设置为"0",就像一些人在其他帖子中提到的那样,但没有成功。
我也在第三台笔记本电脑上试了和第二台一样的,也遇到了同样的问题。
有什么问题吗?

omhiaaxx

omhiaaxx1#

回滚到Tensorflow 2.9,它工作正常。
这可能是Tensorflow 2.11的问题,也可能是文档错误。

相关问题