1、首先查找你的电脑gpu配置,然后安装适合的版本。
鼠标右键,点击NVIDIA控制面板,然后点击系统信息,查看信息,然后查找对应版本。
(11条消息) 2021最新:TensorFlow各个GPU版本CUDA和cuDNN对应版本整理(最简洁)_Kakaluotuo的博客-CSDN博客_tensorflow对应的cuda版本
进入虚拟环境 ,然后首先安装对应版本的cudatoolkit和cudnn
conda install cudatoolkit=10.1.168 (注意这里必须把版本写全,不能只写10.1)
conda install cudnn=7.6.5
pip install tensorflow-gpu==2.3.0 或者 conda install tensorflow-gpu=2.3.0
等待安装完毕