tensorflow-GPU支持

1. 安装N卡驱动
2. 使用conda创建虚拟环境
3. 安装conda依赖包

# 使用 conda 安装 cuda 和 cudnn
conda install cudatoolkit=11.3.1 cudnn=8.2.1

# 自 2.11.0 之后 tensorflow 已经默认包含对 GPU 的支持
pip install tensorflow

# 查看是否检测到GPU
print(tf.config.list_physical_devices(device_type='GPU'))

# 2.12.0 之后不支持直接获取本地 GPU, 需要安装如下插件, 安装完成后训练时会自动使用 GPU
# 否则 tensorflow 代码会报错: \tensorflow-plugins\tfdml_plugin.dll not found
# 安装过程中如果中途报 ERROR 也没关系, 最后一行显示 successfully 就行了
pip install tensorflow-directml-plugin