在windows10上配置tensorflow-gpu!!!
环境:
1. cuda8.0
2. cudnn5.1
3. tensorflow-gpu==1.2.0
4. keras==2.1.1
cuda8.0+cudnn5.1!!!别问为什么,就是好用!!!
首先,将电脑上的nvidia
卸载干净。百度一下很多的可以参考!!
然后,就直接下载cuda8.0和cudnn5.1!!!
下载链接,百度网盘:
链接:https://pan.baidu.com/s/1OR4laGCavSoMoytdkoEZ2g
提取码:fdjn
1、cuda安装
默认路径,点ok直接安装。
自定义安装。
下面other和driver都不打勾!!!!
有Visual Studio的,这个不打勾!!!
2、cudnn
将cudnn
文件夹解压,里面的三个bin、include、lib
复制粘贴到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
下面。。
3、安装tensorflow-gpu&keras
可以镜像加速下载!!
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.2.0
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple keras==2.1.1
搞定!!!!