目录

一、tensorflow安装与测试

1.1 tensorflow 与cuda版本对应版本

1.2 测试gpu

一、tensorflow安装与测试

1.1 tensorflow 与cuda版本对应版本

        官网链接：从源代码构建  |  TensorFlow，选择英文版本：

        本机为ubuntu18.04，cuda为11.0，对照安装tensorflow版本为2.4.0

pip install tensorflow==2.4.0

         注：tensorflow好像在1.15版本及更早区分gpu和cpu版本

https://www.tensorflow.org/install/gpu

1.2 测试gpu

1. python
2. import tensorflow as tf
3. tf.test.is_gpu_available()

        结果是true

         当版本不对时，比如说前一次pip install tensorflow默认安装了tensorflow-2.8.0，结果就找不到gpu（输出False）

@meng