tensorflow

查看NVIDIA基本信息

命令：nvidia-smi

确定cuda版本，可向下兼容，查看目前已安装：

命令：nvcc -V

进入tensorflow官网查看安装

Build from source on Windows  |  TensorFlow (google.cn)

利用anaconda安装

可以看到，tensorflow2.9对应的CUDA版本是11.2，cuDNN版本是8.1

先安装cudatookit

conda install -c conda-forge cudatoolkit=11.2

后安装cudnn

conda install -c conda-forge cudnn=8.1.0

最后安装tensorflow

pip install tensorflow-gpu==2.9.0

验证是否安装成功

在环境中利用下面几条指令验证：

python

import tensorflow as tf

tf.__version__

tf.test.is_gpu_available()

安装ipykernel

方便在jupyter中使用

pip install ipykernel