cuda:https://developer.nvidia.com/cuda-downloads
cudnn:https://developer.nvidia.com/rdp/cudnn-download
cuda,cudnn安装后添加如下环境变量,nvcc -V看是否成功
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\extras\CUPTI\lib64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.4\bin\win64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.4\common\lib\x64
适用于tensorflow 2.6,keras2.6