通过 nvidia-smi 查看到的 CUDA version 为系统最大可装的 CUDA的版本

安装tensorflow-gpu

• 首先确定tensorflow和python的版本（python3.6支持多数版本，是个不错的选择）

• 查看tensorflow对应的tensorflow-gpu所需要的cuda和cudnn版本
  https://www.tensorflow.org/install/source#gpu

创建环境 tf_gpu

• conda create -n tf_gpu python=3.6
• conda install cudnn=7.6.5
• conda install cudatoolkit=10.1
• conda install tensorflow-gpu=2.1

python 检测

import tensorflow as tf
print(tf.__version__)
print(tf.test.is_gpu_available())
print(tf.config.list_physical_devices('GPU'))