会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
LOADING . . .
qizhou
博客园
首页
新随笔
联系
管理
2020年11月16日
Pytorch显存动态分配规律探索
摘要: 下面通过实验来探索Pytorch分配显存的方式。 实验 显存到主存 我使用VSCode的jupyter来进行实验,首先只导入pytorch,代码如下: import torch 打开任务管理器查看主存与显存情况。情况分别如下: 在显存中创建1GB的张量,赋值给a,代码如下: a = torch.ze
阅读全文
posted @ 2020-11-16 23:21 颀周
阅读(1457)
评论(0)
推荐(2)
编辑
公告
很高兴能帮到你~
点赞