从TensorFlow-GPU避坑开始,开启博客之旅,~~~

*关于学习的一切,零零散散的记在jupyter notebook上,恐怕自己都懒得翻看,还是归置下得好,于是乎,终于开始在CSDN上写了第一篇blog,记下关于python,关于机器学习,关于深度学习的一切,大牛都是从菜鸟开始的~~~~*

安装TensorFlow-Gpu的坑

额,不带gpu的TensorFlow玩深度学习,估计是够呛,TensorFlow-gpu必须配套cuda和cudnn,cuda,cudnn网址可以百度一下,

1.先看笔记本显卡是否支持cuda, 在控制面板里面找到:NVIDIA控制面板,系统信息,组件(NVCUDA.dll 后面会显示CUDA的版本号) 能够找到NVCUDA.dll 组件,就代表支持CUDA,或者上https://developer.nvidia.com/cuda-gpus查询

2.下载cuda和cudnn,一定要注意版本的对应,比方cuda选择10.0,cudnn就要选择对应10.0版本的(注意for 后面的cuda版本号),当然要根据自己的操作系统,TensorFlow的版本要求一定要注意,1.13版以后的,cudnn需要7.4以上版本,cuda一般都会下载9.0或者10.0

3.等待cuda安装完毕之后,cudnn文件解压,里面文件夹下的每个文件都要copy到cuda里面对应位置文件夹下面(不要覆盖掉原文件夹)

4.cmd里面,pip install TensorFlow(自己装的是Anaconda,很方便),安装完成之后重启,如果下载过慢,切换到清华镜像,或者直接从清华源里面下载whl文件,用cmd安装

5,试着import TensorFlow,如果报错,注意查看报错信息,让换什么版本的就下载什么版本的,本人就是下错了,装了一下午才弄好,版本对应上了,是可以正常使用(报错缺了什么文件,可以先行在对应文件夹下面搜索是否存在,如果有的话,看下cuda是否添加环境变量,没有的话要重新换cuda版本)

你可能感兴趣的:(深度学习,TensorFlow)