0
点赞
收藏
分享

微信扫一扫

深度学习(四):tensorflow框架搭建

其生 2022-03-15 阅读 91

目录

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

 

 

举报

相关推荐

0 条评论