解决RuntimeError: CUDA error: out of memory

解决RuntimeError: CUDA error: out of memory

跑代码时出现这个错误，输入以下命令

$ nvidia-smi -l

image-20210311211007994

原来是卡0内存爆了，于是指定相对空闲的2卡

CUDA_VISIBLE_DEVICES=2 python my_script.py

可以正常运行

posted @ 2021-03-11 21:12  NTS100K  阅读(1618)  评论(0)  编辑  收藏  举报
浏览器标题切换end