Ubuntu 20.04 安装 tensorflow-gpu

前言

如果非要用最原始的方法去安装tensorflow，不得不说，那还是相当的有点麻烦的。环境配置好了，实验的兴趣就没有了。

操作

1. 不管是pytorch，还是tensorflow，最好还是在anaconda虚拟环境中安装，以做到一个项目一个环境，毕竟有的项目的环境挺恶心人的。

2. 在创建好的一个conda环境中，直接一键命令安装

• 创建环境 （先看下面的提醒，版本对应关系）

  conda create -n test-tf1.9.0 python=3.6
  

• 激活环境

  conda activate test-tf1.9.0
  

• 安装

  conda install tensorflow-gpu=1.9.0
  

• 测试代码

  import tensorflow as tf
  print("GPU Status:", tf.test.is_gpu_available()
  

3. 上面的安装会安装对应的cuda 和 cudnn，所以无需过多担心。

额外提醒：

装之前，最好查找下这个表格（链接见下），看看tensorflow版本和python版本的对应关系

参考：https://blog.csdn.net/baishuiniyaonulia/article/details/118977952