Tensorflow
JaedongXue
这个作者很懒,什么都没留下…
展开
-
tensorflow2总是爆显存的问题
最近在研究tensorflow2,在win环境下装了gpu版本,效率是高很多,但是出现几次爆显存的情况,就是显存被占满了,甚至机子直接黑屏的情况,查了一下资料,tensorflow默认是分配全部显存的,所以会导致显存爆掉。当然可以通过设置来调整:import osos.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"实测,设置之后确实不存在爆显...原创 2020-03-12 15:46:08 · 5472 阅读 · 6 评论 -
win10系统极简安装Tensorflow2.1—gpu版本流程
最近继续研究tensorflow,本打算后续上gpu版本的,但是运行demo过程中,实在太慢受不了了,就研究装了一下gpu版本,现在把具体方法记录如下:一、安装条件可以参考tensorflow官网需求https://tensorflow.google.cn/install/gpu1、nvidia gpu驱动必须在 418.x或者更高,可以用nvidia-smi命令查看2、cud...原创 2020-02-21 20:34:48 · 4866 阅读 · 4 评论 -
Tensorflow 2.1 cpu版本安装问题
tensorflow从去年2019年发布2.0以后,大大改善了api的使用方便性,之前还打算用pytorch,目前考虑到tensorflow的易用性以及tensorflow方便部署到终端的特性,还是好好研究tensorflow。最近一直在研究,遇到问题记录一下,方便自己查阅。1、官方源龟速问题使用豆瓣的源可以解决pip install tensorflow -i https://py...原创 2020-02-20 19:56:38 · 4428 阅读 · 4 评论