cuda cudnn tensorflow对应_【技术茶话会】如何在电脑上搭建 tensorflow 环境

如何在电脑上搭建tensorflow环境

TensorFlow 是一个由谷歌开发和维护的基于数据流编程的符号数学系统，被广泛应用于各类机器学习，完成版的支持 c 和 python 语言，但机器学习模块大多使用 python 语言编写，下面来介绍一下如何在自己的笔记本电脑上搭建 python 语言的 tensorflow 环境。

No.1

首先要在电脑上安装 python，下载地址：https://www.python.org/downloads/

直接下就好了，注意这里要下 3.7 版本的，3.8 不太完善。

No.2

接着就开始安装 tensorflow 了，注意就是 windows 上用的比较多的有两种版本，一个是普通的 cpu 版本，另一个是包含 gpu 加速的版本，gpu 版本的速度大概能快 cpu 版本的 3 倍，同时 cpu 版的安装也较为简单，所以我们下面就着重讲一下 tensorflow-gpu 怎么安装。(要注意 cpu 版本和 gpu 版本不可在一台电脑上同时安装，所以安装前先想好要装哪一个版本)

接着是到英伟达官网下载 CUDA 和 cudnn(CUDA 是一种由 NVIDIA 推出的通用并行计算架构，该架构使 GPU 能够解决复杂的计算问题。NVIDIA cuDNN 是用于深度神经网络的 GPU 加速库。一个是英伟达搭建的 gpu 运算平台，另一个是深度神经网络运算工具包，两者都是要使用 gpu 加速必须要安装的软件。)

安装 CUDA 下载地址：https://developer.nvidia.com/

安装 cudnn 地址：https://developer.nvidia.com/rdp/cudnn-download

注意！这里非常要在意的一点就是 CUDA 和 cudnn 两个版本要对应，不然就安装出错，很烦人，要重新下载安装，所以安装之前一定要好好看清。

不过英伟达官网加载和下载速度是真的慢，可能要开加速器才能好点？显卡不是英伟达的同学。。那就没法子了(捂脸)

No.3

下载完成后要配置 cudnn 环境

1.添加环境参数

2.将 cudnn 添加文件至 CUDA 安装文件夹

No.4

这些都完成后就比较轻松了

下一步安装 anaconda3

下载地址：

https://www.anaconda.com/distribution/#download-section

直接下就好了，这里可以看到只支持3.7的python

安装完成后运行 Anaconda Prompt 这个程序，输入 conda install pip 配置 pip 文件，然后打开电脑的cmd命令行，输入pip install tensorflow-gpu

最后等待安装完成就完事了。

安装完成后我们就可以愉快的和机器学习的算法玩耍了，可以去 github 或 csdn 上自己学习相关算法啦。

比如图像风格转换：

＋

＝

图片和部分资料来自网络，

如有侵权，请联系后台工作人员