错误原因:
使用了CUDA_VISIBLE_DEVICES指定了空闲的3090显卡,但仍报错,原因是batchsize设置得太高,设置了128.
解决方案:
降低到64即可。
或者使用两张及以上的卡。
RuntimeError: CUDA out of memory. Tried to allocate 64.00 MiB (GPU 0; 10.76 GiB total capacity; 7.64
阅读 82
2022-02-27
使用了CUDA_VISIBLE_DEVICES指定了空闲的3090显卡,但仍报错,原因是batchsize设置得太高,设置了128.
降低到64即可。
或者使用两张及以上的卡。
相关推荐
精彩评论(0)