2019年5月4日
TensorFlow中的显存管理器——BFC Allocator
摘要： 背景 作者：DeepLearningStack，阿里巴巴算法工程师，开源TensorFlow Contributor] 欢迎大家关注我的公众号，“互联网西门二少”，我将继续输出我的技术干货~ 使用GPU训练时，一次训练任务无论是模型参数还是中间结果都需要占用大量显存。为了避免每次训练重新开辟显存带来 阅读全文
posted @ 2019-05-04 23:00 DeepLearningStack 阅读(9066) 评论(0) 推荐(8) 编辑