Python 如何解决'；运行时错误：CUDA错误：内存不足'；？

我导入torch并使用

cuda

运行简单代码，但

RuntimeError:cuda error:out out memory

发生错误

我的代码和错误是：

>>> import torch
>>> print(torch.rand(1, device="cuda"))
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
RuntimeError: CUDA error: out of memory

导入火炬 >>>打印（torch.rand（1，device=“cuda”）） 回溯（最近一次呼叫最后一次）： 文件“”，第1行，在 运行时错误：CUDA错误：内存不足

---

我与

torch一起工作

遇到了同样的问题，问题通过以下代码解决：

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '2, 3'

---

确保没有其他进程正在使用您的GPU，如果它们正在使用，请杀死它们或让它们完成。否则重新启动。否则，您的GPU没有足够的内存来支持Pytorch通过“watch-n 0.5 nvidia smi”检查您的GPU内存是否有其他进程使用GPU内存？