本人安装GPU版本的Tensorflow的流程
本人电脑配置:
- win10 + gtx1060
安装版本:
- python3.5
- cuda8
- cudnn6
1. 安装python3.5
直接安装ancoda, 在cmd下 输入python 即可看到版本号
2. 安装coda8
直接安装。按下图选择即可。
3. 安装cudnn6
下载后解压,把对应文件夹下的内容拷贝到cuda/v8.0文件夹下的对应文件夹中,如图示。
4. 环境变量
同时还需将bin目录、lib\64目录添加到环境变量path中,否则后面调用时会报错。
5.安装tensorflow
pip install tensorflow-gpu
网速慢的同学,下载离线包进行安装:http://blog.csdn.net/jacke121/article/details/78870689