AI人生
溶月皑雪
这个作者很懒,什么都没留下…
展开
-
运行deeplab测试遇到的ModuleNotFoundError: No module named 'deeplab'错误解决
最近彻底决定转型做机器学习和AI了,睡眠时间也比平时一度减少了2小时,会不会秃顶最近的任务是研究一下图像语义分割,自然少不了被引诱去看deeplab系列,刚开始构建就没那么顺利:网上查了一下说需要搞一下环境变量,那就搞呗:vi ~/.bashrc在最后添加一行# added by deeplabexport PYTHONPATH=/APP/allen/models...原创 2018-12-06 15:05:46 · 4131 阅读 · 0 评论 -
tensorflow手动指定GPU以及显存大小
以前我们组就一块显卡,不存在指定设备的问题。近期刚插了一块新的gtx 1080ti,几人公用两块卡来做训练、测试、预测等等,网上找了个方式可以指定使用的设备,并且限定使用的显存大小,还是很有用的,亲测可行,拿过来分享给大家~~~~~为了测试方便,使用了mnist代码,数据集和代码都比较小,比较好控制。mnist代码有点多就不贴了,大家自己找找吧。下面开始正题:一般来说GPU devic...原创 2018-12-26 13:14:09 · 1325 阅读 · 0 评论