Ubuntu 安装 tensorflow2.8+cunda 11.2+cudnn 8.1详细步骤

1、conda 建立虚拟环境，python选择3.8版本，高于3.7即可

2、执行命令：conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0

3、执行以下命令：

mkdir -p $CONDA_PREFIX/etc/conda/activate.d
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/' > $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh

4、安装完成后，执行import tensorflow，可能会出现以下错误，主要是protobuf版本过高导致，重新卸载安装即可

pip uninstall protobuf

pip install protobuf==3.20

 

本文主要参考：

 使用 pip 安装 TensorFlow

 新装的TensorFlow出现protobuf相关错误_星辰辰大海的博客-CSDN博客