前言:很多小伙伴在选择CUDA版本和cudnn版本上有疑问,这里简短的说一下,希望能帮到各位小伙伴,我在网上看到有人说,要根据自己的显卡来选择CUDA,其实是错误的,你可以下载自己所需要的CUDA版本,这只是个驱动而已。CUDA的版本和cundd的版本要对应,不然可能会有不兼容报错。CUDA是显卡驱动程序,cundd是用来加速深度学习训练的库,在训练网络时,使用GPU加速,会调CUDA驱动和cudnn库,安装时两者要联系起来,本文的第二步,就是做这个事。
系统环境:windows10 64位
显卡:GTX1060
CUDA版本:cuda_9.0.176_win10(配套的cudnn一定要是7版本)
下载地址:https://developer.nvidia.com/cuda-90-download-archive
CUDNN版本:cudnn-9.0-windows10-x64-v7(7版本的)
下载地址:https://developer.nvidia.com/cudnn
TensorFlow-gpu版本:TensorFlow1.7(1.6和1.5也支持)
1.下载CUDA,