2019年8月29日

网站mark

摘要: neupy一些有用的文章,模型和代码。 http://neupy.com/pages/home.html 阅读全文

posted @ 2019-08-29 21:23 那抹阳光1994 阅读(112) 评论(0) 推荐(0) 编辑

pytorch的显存释放机制torch.cuda.empty_cache()

摘要: Pytorch已经可以自动回收我们不用的显存,类似于python的引用机制,当某一内存内的数据不再有任何变量引用时,这部分的内存便会被释放。但有一点需要注意,当我们有一部分显存不再使用的时候,这部分释放的显存通过Nvidia-smi命令是看不到的,举个例子: Pytorch的开发者也对此进行说明了, 阅读全文

posted @ 2019-08-29 16:58 那抹阳光1994 阅读(42945) 评论(1) 推荐(2) 编辑

导航