Win10环境+ CUDA9.0 +CUDNN7.0+TensorFlow1.7/1.6/1.5配置(亲测有效)
前言:很多小伙伴在选择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, https://developer.nvidia.com/cuda-90-download-archive 按照自己机器的信息选择,下图是我自己机器信息: 下载完之后直接点击运行,会自行安装