笔记 Windows 搭建Tensorflow-GPU 环境(Anaconda + Tensorflow-GPU+ CUDA+ cuDNN)

1,255 阅读2分钟

今天晚上躺了一晚上的坑,现在写篇文章来总结下自己成功的经历

希望小伙伴们看到这篇文章给点个赞 码字不容易 趟坑艰辛

首先查看一下自己的配置:

                              1)系统win10  64bit

                             

                              2)显卡1050ti                               

                             

首先先去Tensorflow官网上面查看自己的版本要求www.tensorflow.org/install/gpu

这里上面明显写了

cuda是要10.0版本以上的

驱动是要410.x以上的

然后cudnn是要7.4.1以上的


所以第一件事是看一下自己的显卡是否能用CUDAhttps://developer.nvidia.com/cuda-gpus



在这里更新你的驱动www.nvidia.com/Download/in…

上面说了我们需要的驱动是410.x以上的



选好你的设备信息点查找然后更新就好了

(这里我要画圈圈给你们圈起来这个很重要很重要一定要做一定要做不然后面容易踩坑)

上面一定要认真看仔细看不然后面容易踩坑

上面一定要认真看仔细看不然后面容易踩坑

上面一定要认真看仔细看不然后面容易踩坑

准备工作完成,开始安装搭建环境

安裝Anaconda & Tensorflow-gpu

Anaconda下载 https://www.anaconda.com/download/

根据自己需求来


安装好了之后



安装CUDA 和cuDNN

在这里找到对应的CUDA的版本下载记住要下本地版本developer.nvidia.com/cuda-downlo…


在这里找 conda 10.0


下载



安装的时候要注意 不要选精简版


下载之后然后安装即可



配置环境变量 没有的话要自己配置


查看安装是否成功用这个能看到 如果出现以下画面就是成功了

nvcc -V


下面开始安装cuDNN7.3

在这里可以下载cuDNN ,要申请账号developer.nvidia.com/rdp/cudnn-d…



下载押解之后能看到的文件,将这几个文件夹复制到

(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0)


把这几个文件放到cuda这里  改名为cudnn



配置变量环境


cmd环境下 安装Tensorflow

pip install  --upgrade tensorflow-gpu


测试是否成功


选择gpu测试 成功 


在此要谢谢龙龙老师 

全套网易云课堂视频链接:study.163.com/course/intr…