使用 CPU 建立 tensorflow 的时候，运算速度不算太快，而且一但出错，就前功尽弃了，看文章说如果使用 NV 的 GPU 进行运算的话，会提速不少。所以做一下测试。

首先需要安装一些工具包：CUDA ToolKit，从 NV 官方网站下载了 cuda 11 的安装包。

注意：NV 官方网站最新版本是 12.0 但是在使用的时候，会发现 tensorflow-gpu 仅支持 11 版本。

然后将环境中的 tensorflow 移除，重新安装 tensorflow-gpu 版本。

conda remove -n env_name tensorflow
conda install -n env_name tensorflow-gpu

说可以在 conda 中直接安装 CUDA 相关的包。

conda search cudatoolkit

conda install cudatoolkit==xx.xx
conda install cudnn==xx.xx

不太清楚是否可以直接使用这个方法配置环境，因为已经安装的 NV 的包了。

安装完成之后，可以利用 tf.config.list_physical_devices('GPU') 来测试是否能使用 GPU 进行运算。