首页 > 其他 > 详细

极简安装 TensorFlow 2.0 GPU

时间:2020-01-15 15:01:08      阅读:81      评论:0      收藏:0      [点我收藏+]

前言

之前写了几篇关于 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 。

技术分享图片

极简安装 TensorFlow 2.0 GPU

原文:https://www.cnblogs.com/csu-lmw/p/12196670.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!