下面通過(guò)實(shí)驗(yàn)來(lái)探索Pytorch分配顯存的方式。 實(shí)驗(yàn)顯存到主存 我使用VSCode的jupyter來(lái)進(jìn)行實(shí)驗(yàn),首先只導(dǎo)入pytorch,代碼如下: import torch 打開(kāi)任務(wù)管理器查看主存與顯存情況。情況分別如下: 在顯存中創(chuàng)建1GB的張量,賦值給a,代碼如下: a = torch.zeros([256,1024,1024],device= 'cpu') 查看主存與顯存情況
www.dbjr.com.cn/article/1998...htm 2025-5-27