Tensorflow 2.3 未找到 GPU
[英]Tensorflow 2.3 doesn't find GPU
问题描述
由于某种原因，tensorflow 不使用我的 GPU。 当我运行tf.test.is_built_with_cuda()它返回False 。 我安装了 tensorflow-gpu，版本 2.3。 我还安装了 CUDA 10.1 和 cuDNN 7.65 并具有正确的环境变量。
当我在控制台中运行nvcc -V时，它会验证我的 CUDA 是版本 10.1，V10.1.243
当我运行nvidia-smi时，它显示我的驱动程序版本是 460.79，CUDA 版本是 11.2
我是否可能需要降级我的 GPU 驱动程序？
1 个解决方案
解决方案1
0 2021-01-01 07:58:53
我也有同样的问题，但是当我安装 PyTorch 时，它正确地检测到了我的 GPU，之前的tf 2.1.0工作正常。
也许这是一个错误。
我将尝试安装tf 2.1并查看它是否检测到 GPU。
更新 1/7/2021：尝试通过 conda 安装安装 tf 2.1.0 版本，但没有成功。
Tensorflow was able to detect my gpu.I uninstalled the tf from conda installation and installed it through pip .I think the conda version doesn't detect the gpu.Make sure you the tensorflow version you are installing matches the requirement of the CUDA and cudnn图书馆。
我当前的配置： tensorflow-gpu ：2.3.0
CUDA ：7.6.5
cudnn :10.1
tensorflow-gpu 2.3 安装了 cuda 并且 cudnn 无法检测到 GPU？
[英]tensorflow-gpu 2.3 install with cuda and cudnn can't detect GPU?
声明:本站的技术帖子网页，遵循CC BY-SA 4.0协议，如果您需要转载，请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.