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

目录

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

python
import tensorflow as tf
tf.test.is_gpu_available()

        结果是true

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

@meng