windows10安装cuda11.4,cudnn

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

posted on 2021-12-24 20:23  medsci  阅读(630)  评论(0编辑  收藏  举报