tensorflow安装版本问题

TensorFlow安装问题解决

查看安装版本信息

• TensorFlow与cuda对应关系
  

• cudnn 和 cuda版本对应
  

创建Anconda环境

自己创建的环境配置如下：

环境名称	Python版本	tensorflow-gpu版本	cuda版本	cudnn版本
TensorFlow	3.9	2.10.0	11.2	8.2.1

• 创建环境 conda create -n TensorFlow python=3.9
• 激活环境 conda activate TensorFlow
• 可以在anaconda环境中安装对应cuda版本 conda install cudatoolkit=11.2 （查看要下载的TensorFlow支持的cuda版本）
• 安装对应cudnn conda install cudnn=8.2.1 （查看上一步下载的cuda版本对应的cudnn版本）
• 安装TensorFlow pip install tensorflow-gpu==2.10.0 -i https://pypi.mirrors.ustc.edu.cn/simple（根据自己选择的版本写）
• 测试：输入python 进入python环境，导入tensorflow包import tensorflow as tf,没有报错（多余信息）即成功

总结
TensorFlow有时需要额外下载低版本的cuda，为了避免在本地下载多个版本的cuda造成混乱，可以利用Anaconda，在创建的TensorFlow的虚拟环境中下载对应版本的cuda和cudnn，这样就可以成功使用gpu，也不影响别的环境。