在win10下安装Tensorflow1.3-gpu(CUDA8

2017-09-06  本文已影响0人  给力哥

前提:安装的是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

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

上一篇下一篇

猜你喜欢

热点阅读