一、确定环境
第一,明确是在Windows环境还是linux环境安装;
第二,明确本机有没有安装cuda和cudnn,若没有,请参考:CUDA和CUDNN安装;
第三,明确自己的Python版本,版本不同有较大影响。以Windows环境为例,说明这几者的关系。
该图来自其安装说明,请务必确保自己的环境能够对应起来。
第四,不建议用太新版本的python来安装tensorflow,个人推荐3.6-3.10。
第五,从TF 2.11开始,Windows不支持CUDA构建。要在Windows上使用TensorFlow GPU,试试安装2.10.0版本。
pip install tensorflow-gpu==2.10.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/
二、安装教程推荐
由于网上已经有非常多的教程帖,这里不再赘述,推荐几个。
- tensorflow-gpu安装教程(面向小白+踩坑记录)
- tensorflow_gpu安装
- tensorflow安装官方教程中文版
- Installing TensorFlow 2 GPU [Step-by-Step Guide]
三、报错问题
安装时遇到报错是正常的,请先百度、谷歌搜索解决办法,若未能解决,请在社区发帖留言,详细描述环境版本、安装命令及遇到的报错。