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

本文主要参考:
版权声明:本文为qq_18256855原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。