如何安装tensorflow-gpu？

Question

我安装了库达-11.0和cuDNN-8.2和张量-RT 7.1.然后安装了tensorflow-gpu。但是tf.is_gpu_available()显示出了错误。实际上，我先安装了tensorflow-gpu。我有pip3 20.1.1和Nvidia GForce 930MX和Ubuntu20.04。

下面是nvidia-smi命令输出：

Answer 1

目前，Tensorflow 2.1不支持CUDA 11.0。将CUDA降级到10.1版，然后再试一次。

参考资料：https://www.tensorflow.org/install/gpu

Answer 2

对于macosx/linux安装- 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

Answer 3

Tensorflow无法找到gpu，因为Cuda工具包版本是11，但是，cuDNN版本是8.2RC + cuda 11。但是在Tensorflow-2.2.0中，当cuda工具箱和cudnn都有10.1版本的cuda时，它就可以工作了。不支持Cuda 11.0。