首先注意:TensorFlow2.11开始就不支持Windows的GPU,若要在Windows下安装GPU版本的,只能安装2.10及以下的版本。
下面是官网提示直接安装,直接安装的话是最新版本的:
-
若要安装指定版本,使用命令:
pip install tensorflow[and-cuda]==2.xx
-
注意安装版本要和python版本以及cuDNN、CUDA版本要对应,可去官网查:
https://tensorflow.google.cn/install/source
-
安装cuDNN以及CUDA命令:
conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0