安装CUDA
1. 准备好NVIDIA的显卡，下载安装CUDA：https://developer.nvidia.com/cuda-downloads
2.安装好之后把CUDA安装目录下的bin和lib\x64添加到环境变量Path中
安装cuDNN
1. cuDNN下载：https://developer.nvidia.com/rdp/cudnn-download
2. 解压压缩包，把压缩包中bin，include，lib\x64中的文件分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
3. 把C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64\cupti64_101.dll拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
然后卸载原来版本的tensorflow
pip uninstall tensorflow
安装gpu版本的tensorflow
pip install tensorflow-gpu