我安装了cuda-11.0和cuDNN-8.2和Tensor-RT 7.1。然后安装tensorflow-gpu。但tf.is_gpu_available()显示False。实际上，我先安装了tensorflow-gpu。我有pip3 20.1.1和Nvidia GForce 930MX和Ubuntu 20.04。
这是nvidia-smi命令输出：
如何安装 tensorflow-gpu？
数据挖掘
张量流
显卡
2022-02-25 15:24:45
3个回答
目前 Tensorflow 2.1 不支持CUDA 11.0。将 CUDA 降级到 10.1 版，然后重试。
用于 macosx/linux tensorflow gpu 安装 - https://www.youtube.com/watch?v=MpUvdLD932c&list=PLjy4p-07OYzulelvJ5KVaT2pDlxivl_BN&index=8。对于 Windows 10 - https://www.youtube.com/watch?v=qrkEYf-YDyI&list=PLjy4p-07OYzulelvJ5KVaT2pDlxivl_BN&index=9
Tensorflow找不到 gpu，因为Cuda Toolkit版本是 11。但是，cuDNN版本是 8.2RC + cuda 11。但是在Tensorflow-2.2.0中，当 cuda toolkit 和 cudnn 都具有 10.1 版本的 cuda 时，它可以工作。不支持 Cuda 11.0。
其它你可能感兴趣的问题