我的tensorflow-gpu安装真实避坑指南

tensorflow-gpu安装真实避坑指南2019-12-04

我的电脑配置:

微星GL63笔记本

CPU Inter Core i7 8代

GPU NVIDIA GeForce GTX1060

Windows10


本篇文章适合电脑上没有python,也没有任何相关软件的小白,大神请绕道~

自己查遍了网上的文章,前前后后装了删删了装无数遍后摸索出来的。

需要安装的有:

Anaconda3(64bit)

CUDA-9.0

CuDNN-7

Python-3.6

TensorFlow-gpu

pycharm

一、安装Anaconda

直接官网最新版本下载,注意32还是64的。anaconda自带python,故不用之前自己安装python,有装过的先卸载干净,环境变量也删掉。

装的时候一定要2个都勾选,让软件自己添加环境变量

二、安装cuda和cudnn

安装之前先更新自己的显卡驱动https://www.nvidia.cn/Download/index.aspx?lang=cn

查看自己的cuda版本号,在https://developer.nvidia.com/cuda-toolkit-archive选择,注意不一定下载对应版本,我的笔记本是10版本,我下载的是9版本,过新的版本可能不支持,本人也试过无数次,最终确定9版本。但是注意,电脑支持的版本要大于你下载的版本。记好安装路径。

装好后在dos窗口中输入nvcc -V,如果输出是cuda版本信息,则说明安装成功


我下载的cuda9

下载cudnn,注意需要注册一下https://developer.nvidia.com/rdp/form/cudnn-download-survey,选择对应的版本。下载后解压看到三个文件,将这三个文件夹复制到安装CUDA9.0的路径覆盖原有的文件。

之后配置环境变量:桌面时右键“此电脑”,选择“属性”打开系统窗口,然后选择并打开“高级系统设置”进入系统属性窗口,打开环境变量,然后将C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64,这三个路径填到系统变量的Path中,这样环境变量就配置好了。

三、安装tensorflow

打开Anaconda Prompt

直接输入:

conda create -n tensorflow_gpu python=3.6 ,用来创建Tensorflow环境。中间按y回车确认。

装好后输入activate tensorflow_gpu进入创建好的环境,输入conda install anaconda安装一些包

装好后输入pip install --upgrade tensorflow-gpu==1.12.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

安装tensorflow的gpu1.12版本

四、验证

打开Anaconda Prompt输入activate tensorflow_gpu进入刚创建的环境,输入python进入编译,输入:

import tensorflow as tf

hello = tf.constant('Hello, TensorFlow!')

sess = tf.Session()

print(sess.run(hello))

输出:

b’ Hello, TensorFlow!


恭喜你安装成功!

五、关联pycharm

我是用pycharm跑代码的,也可以用别的。

安装pycharm,最好用专业版,安装教程https://blog.csdn.net/mars_xiaolei/article/details/89510741

装好后新建工程

Location任意,选择Existing interpreter,选择刚才Anaconda中创建的tensorflow_gpu下的prthon.exe,这样就关联上啦,可以用gpu跑代码了。

装了2天,不想让更多的人浪费时间安装,所以第一次写文章,有不妥当的地方和错误欢迎指正~

下面附上一些有帮助的文章:

https://www.jianshu.com/p/62f155eb6ac5

https://www.cnblogs.com/gaofighting/p/9917456.html

https://blog.csdn.net/AAlonso/article/details/81504036



图像识别小记

需要 pillow 和 pytesseract 这两个库,进入tensorflow_gpu空间pip install 安装就好了。

还需要安装 Tesseract-OCR.exe 然后配置下就好了。

在https://digi.bib.uni-mannheim.de/tesseract/下载Tesseract-OCR,我下载的4.0.0然后配置环境变量,将软件的安装路径添加到环境变量的path中,配置完成后win+r输入cmd命令行输入tesseract -v,如果出现如下图所示,说明环境变量配置成功。

在Anaconda目录中搜索pytesseract.py并打开,找到 tesseract_cmd,改变它的值为刚才安装 tesseract.exe 的位置,使用r进行不转义处理。

然后就可以进行简单的图像识别啦,可以识别数字,英文和中文,只要添加了相应的语言库支持,添加方法如下https://blog.csdn.net/sherpahu/article/details/80380373

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