ubuntu系统tensorflow框架搭建与测试 —— 深度学习(四)

一、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）