极简安装 TensorFlow 2.0 GPU

前言

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

不过好在去年下半年 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.0 python=3.7

 然后

activate tf2.0
conda install tensorflow-gpu==2.0

记住,这里一定是 conda Install

 

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

 好的,成功!

然后就是愉快的代码时间

 

 小节

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

猜你喜欢

转载自www.cnblogs.com/csu-lmw/p/12196670.html