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

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. 原因 内存不足，对于我的情况来说，是虚拟内存不足（硬盘没空间了）。虽然物理内存是有的，但是虚拟内存没了，看下图 已提交 可以看到增加