摘要: 解决RuntimeError: CUDA error: out of memory 跑代码时出现这个错误,输入以下命令 $ nvidia-smi -l 原来是卡0内存爆了,于是指定相对空闲的2卡 CUDA_VISIBLE_DEVICES=2 python my_script.py 可以正常运行 阅读全文
posted @ 2021-03-11 21:12 NTS100K 阅读(1620) 评论(0) 推荐(0) 编辑
浏览器标题切换end