记录要跑的text to image,一步步不要太精细

首先,我们需要准备下载text  to image 资源,我打算跑的代码地址

要求的版本

然后查看自己的服务器cudn,tensorflow,NLTK,如图

1.tensorflow版本

python

import tensorflow as tf

tf.__version__

如果不报错并显示版本,则安装正确,否则按照安装教程这个链接自行安装。

tf.__path__

路径

直接输入上面的命令,得到这个结果

2.tensorlayer版本

tensorlayer tensorlayer安装官方推荐运行在python3上,看上图知道,这是在2.7上。最便捷安装 TensorLayer 只需要一个指令。如下

sudo pip install git+https://github.com/zsdonghao/tensorlayer.git

执行上步骤安装

1.7.3版本

3.安装NLTK

sudo pip install -U nltk 即可 

在python 的终端里面,输入:import nltk 如果不报错即表明安装成功.然后

4.下载源码

然后clone源码到自己环境里。

sudo git clone https://github.com/zsdonghao/text-to-image.git

等待完成。一秒秒就好了。

5.数据集的下载

我在python2下,有弄module tqdm的问题,所以我用这个命令:pip install tqdm 

然后执行python downloads.py下载数据集,封装了,真好呀!然而,需要翻墙的奥!

6.下好之后,执行data_loader.py,对数据经行处理。

python data_loader.py

7.训练模型

 python train_txt2im.py

我能说我们实验室都在跑程序吗!

胖程序出错

使用下面的命令查看使用情况

nvidia-smi

使用情况100%

…………………………

ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[64,128,64,64]

…………………………

这个错误一出现。

1.*首先得排除整个显卡上是不是有别的模型在跑?

2.如果没有,检查自己的模型逻辑有没有错误?(很多时候,即使没错也会崩……)

3.*接下来,在进行下一步尝试……

先用invidia-smi命令,看看服务器上有哪些空闲显卡(本地电脑可忽略,重新运行就是了).

对于tensorflow来说,可以用CUDA_VISIBLE_DEVICES=1(,2...) python train.py​,命令来运行程序,其中的数字1,2...代表显卡序号。如果你觉得一个显卡不够,或是想让程序跑得快,在不影响别人使用的情况下,可以多选择几块显卡。CUDA_VISIBLE_DEVICES命令是用来确定CUDA允许tensorflow发现的显卡​。如果没有,默认情况下tensorflow会利用所有能利用的资源。

对于濒临淘汰的Theano来说,运行时设置device=1(,2,3..),即可。

 4.如果还是不行,请回到第二步……不要过度自信……应引用自别人说的

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏木可大大

漫谈文件系统

翻译成中文大致意思:文件系统主要是管理数据存储以及数据如何检索的,而数据存储在磁盘或内存中。上期我们聊过了漫谈虚拟内存,本期我们就重点介绍磁盘中的机械磁盘的组成...

530120
来自专栏FreeBuf

Nmap 7.50更新:自去年12月来的重大更新

Network Mapper 近日发布了最新更新 Nmap 7.50 ,上一次的更新还需要追溯到2016年12月,而此次的版本上对于对于数百项功能进行了改进。 ...

29340
来自专栏AI研习社

Github 项目推荐 | 基于 web 的视频图像标记工具 CVAT

CVAT 是加州尔湾视频标注工具(http://carlvondrick.com/vatic/)的重设计和重实现版本。它是用于计算机视觉的免费在线交互式视频和图...

22830
来自专栏FreeBuf

NSA武器库之Eternalchampion(永恒冠军)复现

准备工作 攻击机1:带有漏洞利用工具集的XP,并且此次需要准备好WinHex, IP 172.26.97.35 攻击机2:Kali, IP 172.26.97....

28960
来自专栏LhWorld哥陪你聊算法

【深度学习篇】--Windows 64下tensorflow-gpu安装到应用

一直以为自己的笔记本不支持tensflow-gpu的运行,结果每次运行模型都要好久。偶然间一个想法,想试试自己的笔记本,结果竟然神奇的发现能用GPU。于是分享一...

20910
来自专栏MYSQL轻松学

Mysql Group Replication介绍

一、Mysql Group Replication简介 Mysql Group Replication(MGR)是一个全新的高可用和高扩展的MySQL集群服务...

71440
来自专栏jeremy的技术点滴

锁定NodeJS项目的依赖库

30770
来自专栏FreeBuf

PHP任意文件上传漏洞(CVE-2015-2348)

安全研究人员今天发布了一个中危漏洞——PHP任意文件上传漏洞(CVE-2015-2348)。 在上传文件的时候只判断文件名是合法的文件名就断定这个文件不是恶意文...

30850
来自专栏吴伟祥

Linux Partition scheme 分区方案(一)

根分区包含Linux系统所有的目录。如果在安装系统时只分配了/分区,那么上面的/boot、/usr和/var将都包含在根分区中,也就是这些分区将占用根分区的空间...

29520
来自专栏解Bug之路

解Bug之路-记一次JVM堆外内存泄露Bug的查找 顶

JVM的堆外内存泄露的定位一直是个比较棘手的问题。此次的Bug查找从堆内内存的泄露反推出堆外内存,同时对物理内存的使用做了定量的分析,从而实锤了Bug的源头。笔...

10240

扫码关注云+社区

领取腾讯云代金券