用只有2个G的显卡跑数据就需要在训练之前先把无关进程杀掉,防止跑到一半显存满了
nvidia-smi
:显示当前GPU中的线程
kill -9 PID
:输入PID以结束线程
用只有2个G的显卡跑数据就需要在训练之前先把无关进程杀掉,防止跑到一半显存满了
nvidia-smi
:显示当前GPU中的线程
kill -9 PID
:输入PID以结束线程
本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。