前言
这篇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的支持,所以我们可以先下载安装它们。
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(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版本)