2021年6月24日
解决“RuntimeError: CUDA out of memory.”
摘要： 1.可以在终端里面使用python进入python环境 >>>import torch >>>torch.cuda.empty_cache() 2.使用nvidia-smi查看gpu使用进程，找到进程的名字，使用sudo pkill python去杀死他 3.减少batch size的大小，最好是8 阅读全文
posted @ 2021-06-24 21:36 小Aer 阅读(47) 评论(0) 推荐(0) 编辑