我无法配置tensorflow与GPU一起使用

问题描述 投票:0回答:1

我使用的是 lenovo l340 15 irh 笔记本电脑,它具有独立的 nvidia 1650 显卡。 操作系统:Windows Python版本:3.9.13 IDE:Visual Studio 代码

我已经安装了CUDA、cuDNN,但是代码

len(tf.config.list_physical_devices('GPU'))>0

继续返回False。

为了调查这个问题,我在 Windows 命令行上写了

nvidia-sli
并得到了以下结果: nvidia-smi

然后我写了

nvidia-smi --query-gpu=compute_cap --format=csv
并得到了这个: nvidia-smi --query-gpu=compute_cap --format=csv

然后我发现了我应该使用哪些版本。我的tensorflow版本是2.14.0,所以最新版本的CUDA不适合这个tensorflow。我卸载了它并安装了 CUDA v11.8 但没有帮助。

然后我安装了 cuDNN,它支持 CUDA v11.x,但也没有帮助。

我还添加了 cuDNN v11.x 和 v12.x 的系统变量“Path”实际路径: Path variable

现在我陷入困境,不知道下一步该做什么。你能帮我吗?

tensorflow gpu
1个回答
0
投票

Kindy 使用此链接软件和硬件安装所需的硬件和软件要求。并按照分步说明进行操作。

安装完所有必需的软件和硬件要求后,请按照以下步骤操作。

系统要求:

Windows 10 19044 或更高版本(64 位)。这对应于 Windows 10 版本 21H2(2021 年 11 月更新)。

GPU 设置:

如果尚未安装 NVIDIA GPU 驱动程序,请安装。您可以使用以下命令来验证它是否已安装。

nvidia-smi

安装 TensorFlow :

TensorFlow 需要最新版本的 pip,因此请升级您的 pip 安装以确保您运行的是最新版本。

pip install --upgrade pip

然后,使用 pip 安装 TensorFlow。

 # For GPU users
 pip install tensorflow[and-cuda]

验证安装:

验证 GPU 设置:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

如果返回GPU设备列表,则说明您已成功安装TensorFlow。

按照上述步骤安装tensorflow-gpu。

© www.soinside.com 2019 - 2024. All rights reserved.