基于conda环境的tensorflow-gpu环境的搭建

前言

​ 这篇blog是基于安装好了anaconda环境下的tensorflow环境的搭建。所以这里安装anaconda不再赘述。

​ 可以先查看一下自己的anaconda环境搭建好了没有：

​ 然后，安装tensorflow可以选择CPU方式或GPU方式，GPU方式对矩阵运算有很好的支持，会加速程序的执行，GPU版本需要CUDA和cuDNN的支持，所以我们最好要有一张Nvidia的显卡，我自己的电脑是装了GTX1050显卡，在这里查看自己的电脑是否支持。

正式安装

​ 打开tensorflow官网：https://www.tensorflow.org/install#installing_with_anaconda

可以先为conda添加国内镜像：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --set show_channel_urls yes

安装CUDA和cuDNN

​ 因为安装tensorflow需要CUDA和cuDNN的支持，所以我们可以先下载安装它们。

在这里查看一下最新版本对应的CUDA和cuDNN：

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-YUDIeclo-1572850067825)(https://img3.doubanio.com/view/photo/m/q6tLCg8G_CAboFHbAzE-AQ/178857354/x2573011020.jpg)]

所以我们就下载CUDA10.0和对应版本的cuDNN。

CUDA安装即可，自己一般都已经安装了Nvidia的驱动，（我喜欢玩玩游戏，所以之前安装了GameDriver），就可以手动选择只安装单独的部分，但是请记住安装cuda的路径。安装完成之后，将cuDNN解压到CUDA的目录即可。

创建一个conda环境

​ 创建一个专门的环境用来使用tensorflow更好一点，不然你为什么还要选择使用anaconda呢？

​ conda create -n tensorflow python=3.6

这里我选择环境的python版本为比较成熟的3.6版本。

之后只需要conda install tensorflow-gpu=2.0即可安装完成。（我选择的是安装最新版本，2.0版本）