运行时错误:CUDA内存不足,pytorch

jpfvwuh4  于 2022-11-09  发布在  其他
关注(0)|答案(1)|浏览(341)

运行时错误:CUDA内存不足。已尝试分配128.00 MiB(GPU 0; 14.76 GiB总容量;已分配10.85 GiB; 27.75 MiB可用空间; PyTorch总共保留了11.31 GiB)如果保留的内存〉〉分配的内存,尝试设置max_split_size_mb以避免碎片。

inb24sb2

inb24sb21#

PyTorch总共预留了11.31 GiB您是否运行了另一个python内核?(如notebook),您可以在终端输入nvidia-smi来检查。

然后如果你想终止这个未使用的进程,你可以通过输入kill -9 <pid number>来完成。使用-9意味着你强制关闭它。

相关问题