CUDA out of memory(CUDA顯存不足)

1.查看CUDA使用情況。 在C:\Program Files\NVIDIA Corporation\NVSMI位置打開命令行,然後輸入nvidia-smi.exe -l 1,可每秒更新CUDA使用情況,如圖所示。 2.本機的CUDA顯存爲2048MiB,當前使用爲67MiB,運行模型後,發現當前CUDA使用情況超過2048MiB,解決方法爲減小batch_size的大小(減小後結果可能會變差),
相關文章
相關標籤/搜索