我正试图用jupyter笔记本运行pytorch代码,但出现了此错误
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 1.23 GiB already allocated; 18.83 MiB free; 1.25 GiB reserved in total by PyTorch)
我已经找到了答案。最重要的是要减少批量大小。 我已尝试将批量大小从20减少到10,再减少到2和1。现在仍然无法运行代码
我的GPU是Quadro p620
您可以通过以下方式释放现有内存:
注意这个过程,并使用
相关问题 更多 >
编程相关推荐