admin管理员组

文章数量:1611555

错误原因:

使用了CUDA_VISIBLE_DEVICES指定了空闲的3090显卡,但仍报错,原因是batchsize设置得太高,设置了128.

解决方案:

降低到64即可。

或者使用两张及以上的卡。

本文标签: allocateMIBRuntimeErrorCUDAMemory