运行时错误:CUDA内存不足。已尝试分配128.00 MiB(GPU 0; 14.76 GiB总容量;已分配10.85 GiB; 27.75 MiB可用空间; PyTorch总共保留了11.31 GiB)如果保留的内存〉〉分配的内存,尝试设置max_split_size_mb以避免碎片。
inb24sb21#
PyTorch总共预留了11.31 GiB您是否运行了另一个python内核?(如notebook),您可以在终端输入nvidia-smi来检查。
nvidia-smi
然后如果你想终止这个未使用的进程,你可以通过输入kill -9 <pid number>来完成。使用-9意味着你强制关闭它。
kill -9 <pid number>
-9
1条答案
按热度按时间inb24sb21#
PyTorch总共预留了11.31 GiB您是否运行了另一个python内核?(如notebook),您可以在终端输入
nvidia-smi
来检查。然后如果你想终止这个未使用的进程,你可以通过输入
kill -9 <pid number>
来完成。使用-9
意味着你强制关闭它。