出现 CUDA out of memory 的解决方法

1. 问题所示

运行这段程序的时候出现如下错误:

RuntimeError: CUDA out of memory. Tried to allocate 1.38 GiB (GPU 0; 7.80 
GiB total capacity; 5.94 GiB already allo

猜你喜欢

转载自blog.csdn.net/weixin_47872288/article/details/127837815