之前做本科毕业设计的时候就用tensorflow做深度学习相关的内容,当时受限于电脑的配置,无法使用tensorflow-gpu,再读研前换了电脑,并准备安装tensorflow-gpu,据说速度特别快,于是按照网上的一些教程,安装了上去,本以为安装好了,然而在一次运行到大的数据时发生了错误才知道竟然一直用的是cpu,并未使用gpu,按照https://blog.youkuaiyun.com/lwhsyit/article/details/89465387第一个方法检查才知道自己没有安装好tensorflow-gpu版本,于是各种方法都尝试了,来来回回折腾了3天,终于安好了,因此想写出来与大家分享,当然可能会不详细也缺少一些介绍,但是会把自己遇到的一些坑都说出来供大家参考。
基本介绍
要想安装tensorflow-gpu首先得知道自己的电脑配置能不能达到要求,我的显卡是NVIDIA的GEFORCE GTX1050TI,可通过https://developer.nvidia.com/cuda-gpus查看。
此外,安装的时候一定要把cuda和cuDNN的对应关系弄清楚,https://blog.youkuaiyun.com/qq_326324545/article/details/91431007这篇文章写的很清楚,我把其中gpu的列出来。
之后就是安装的过程了
Anaconda和tensorflow-gpu的安装
首先安装Anaconda,利用其来安装python3.6,tensorflow-