0

点赞

收藏

分享

RuntimeError: CUDA out of memory. Tried to allocate 64.00 MiB (GPU 0； 10.76 GiB total capacity； 7.64

1kesou 2022-02-27 阅读 82

标签: pytorch 人工智能 python

错误原因：

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

解决方案：

降低到64即可。

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

0 条评论

关注