极简安装 TensorFlow 2.0 GPU

时间:2021-11-13 07:22:19

前言

之前写了几篇关于 TensorFlow 1.x GPU 版本安装的博客,但几乎没怎么学习过。之前基本在搞 Machine Learning 和 Data Mining 方面的东西,极少用到 NN,虽然看过几次相关代码,但没怎么看懂过,静态图是有些复杂,对像我这样的菜鸡来说难度有那么点点点点点大 orz。。。

极简安装 TensorFlow 2.0 GPU

不过好在去年下半年 TensorFlow 2.0 终于出了,前段时间安装了并学习了下,感觉确实要简单了很多,可能是因为我用的 tensorflow.keras 的原因吧。不管怎么说,用的爽就完事儿了。

安装

那就来讲讲安装过程吧,比 TensorFlow 1.x GPU 版本安装要简单不知道多少。

首先还是在 anaconda 中进行,所以,没有 anaconda 的就装一个吧。关于 anaconda 的安装请戳这里

然后我们要设置清华源镜像,好像蛮久之前这个镜像被封了,但是现在似乎又可以用了,详情看这里

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

接下来,开始操作

conda create -n tf2. python=3.7

极简安装 TensorFlow 2.0 GPU

然后

activate tf2.
conda install tensorflow-gpu==2.0

记住,这里一定是 conda Install

极简安装 TensorFlow 2.0 GPU

极简安装 TensorFlow 2.0 GPU

等安装完成后,来测试测试

极简安装 TensorFlow 2.0 GPU

好的,成功!

然后就是愉快的代码时间

极简安装 TensorFlow 2.0 GPU

小节

到这里我们的 TensorFlow 2.0 GPU 就安装好了,这里给大家推荐一个资料,感觉还不错,前段时间就是看着这个学习的。希望这篇教程能帮助到大家,不说了,我接着去改代码跑实验了,在实验室咸鱼了大半年,寒假和下学期基本上就没啥事了,可以认真搞搞科研了,2333333 。

极简安装 TensorFlow 2.0 GPU