在win10下安装Tensorflow1.3-gpu(CUDA8.0+cuDNN6.0+python3.6)

前提:安装的是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
安装完需要设置系统环境变量

image.png

3.安装tensorflow-gpu
1.3版本已经支持python3.6了
只有python3的用
pip install --upgrade tensorflow-gpu
双版本python的用
pip3 install --upgrade tensorflow-gpu

最后打开cmd

image.png

理论上没报错就代表完成了

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容