前提:安装的是GPU版本的tensorflow,CPU的直接安装就行了,不需要第一第二步
写这篇文章的时候,tensorflow最新版已经是1.3,网上的安装教程都是1.3之前的,cuDNN也都还是用的5.1,所以踩了坑,那就记录下来,让后面的朋友少踩点
1.下载安装CUDA8.0
https://developer.nvidia.com/cuda-80-ga2-download-archive
2.下载安装cuDNN6.0,这里不要下5.1了,7.0我自己尝试也是不行的,下载需要登录,耐心点吧
https://developer.nvidia.com/rdp/cudnn-download
安装完需要设置系统环境变量
3.安装tensorflow-gpu
1.3版本已经支持python3.6了
只有python3的用
pip install --upgrade tensorflow-gpu
双版本python的用
pip3 install --upgrade tensorflow-gpu
最后打开cmd
image.png理论上没报错就代表完成了
网友评论