我使用的是 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 --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”实际路径:
现在我陷入困境,不知道下一步该做什么。你能帮我吗?
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。