RuntimeError:CUDA error:out of memory
解决方法:
当前gpu内存不足,换另一块显卡
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1'#默认第0块内存不足时,选择第1块或第2块显卡。
RuntimeError:CUDA error:out of memory
解决方法:
当前gpu内存不足,换另一块显卡
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1'#默认第0块内存不足时,选择第1块或第2块显卡。