[人工智能]深度学习(四)：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）