![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
机器学习
冉冉升起的码农
这个作者很懒,什么都没留下…
展开
-
pytorch API 的使用
一:index_fill_dim 表示的是哪一维, dim = 0, 表示的列; dim = 1 表示行原创 2018-11-16 12:32:08 · 224 阅读 · 0 评论 -
pytorch 减小显存消耗,优化显存使用, 计算模型的中间变量
一、 模型如下:i is 0, m is RFBNet( (base): ModuleList( (0): Conv2d(1, 24, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (1): ReLU(inplace) (2): Conv2d(24, 24, kernel_size=(3, 3), stride...原创 2019-01-16 09:29:28 · 2808 阅读 · 0 评论 -
pynvml 查看当前所使用的显存
import pynvmlpynvml.nvmlInit()# 这里的0是GPU idhandle = pynvml.nvmlDeviceGetHandleByIndex(0)meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)print(meminfo.used)转载 2019-04-19 18:54:56 · 1514 阅读 · 0 评论