OFA-Sys/chinese-clip-vit-base-patch16 占用显存测试

标签：显存 __ OFA chinese workers torch 虚拟内存

model.get_image_features(inputs)  

64 batch_size 2096MB

取消with torch.no_grad():后 8GB占满

16 batch_size 3886MB

 

---

AutoModel .from_pretrained(MODEL_NAME) 执行慢，原因是需要启用网络代理，否则总是卡在验证阶段  

---

DataLoader 增加num_workers后 torch.cuda.OutOfMemoryError: CUDA out of memory. 原因可能是把import放在了if __name__ == '__main__':外边？或者我更新了一下显卡驱动？ 神奇了，过了一天我把代码还原回“增加num_workers后 torch.cuda.OutOfMemoryError: CUDA out of memory.”时的情况，又不报错了    

---

  MemoryError: Caught MemoryError in DataLoader worker process 0. 原因 内存不足，对于我的情况来说，是虚拟内存不足（硬盘没空间了）。虽然物理内存是有的，但是虚拟内存没了，看下图 已提交 可以看到增加

标签：显存,__,OFA,chinese,workers,torch,虚拟内存
From： https://www.cnblogs.com/hhdom/p/18536444