ubuntu安装tensorflow gpu版本最简单流程

最简单流程

使用conda一键安装，但是只能支持到tensorflow-gpu2.4版本

# 不要指定python版本
conda create -n tf
conda activate tf
conda install tensorflow-gpu=2.4

指定版本安装

1.查询支持的tensorflow-gpu版本和对应的python版本、cudatoolkit、cudnn版本Tensorflow、python、keras和cuda的对应版本（更新于2022.6.18） - 知乎 (zhihu.com)

2.创建环境

# 指定python版本
conda create -n tf python=3.8
conda activate tf
conda install pip

3.pip安装指定的tf-gpu版本

# 以2.5版本为例
pip install tensorflow-gpu==2.5

4.conda安装指定的cudatoolkit和cudnn版本

# cudatoolkit和cudnn版本是向前兼容的，可以安装更高版本，但cudatoolkit不能超过nvidia驱动支持的最高版本
# 也就是nvidia-smi输出的右上角显示版本
# 使用conda search cudatoolkit查询conda支持的版本，cudnn同理
conda install cudatoolkit=11.3.1 cudnn=8.2.1

5.检查tensorflow-gpu是否安装成功

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