运行深度学习代码时报错RuntimeError: CUDA out of memory. Tried to allocate 482.00 MiB

在自己的电脑上运行Pointnet++(pytorch版本)时,6g显卡,batchsize设置成8报错

尝试了https://blog.csdn.net/weixin_32393347/article/details/104395179 的方法,仍然不行。

最后只能将batchsize改为4,得以运行成功。

你可能感兴趣的:(python,深度学习,pytorch)