TEXTure torch.cuda.OutOfMemoryError: CUDA out of memory.
时间: 2023-09-21 17:03:42 浏览: 98
您遇到的问题是CUDA显存不足的错误。根据引用中的错误信息,您的GPU总容量是10.92 GiB,已分配了10.10 GiB的显存,还有150.69 MiB的空闲内存,而您的程序尝试分配了858.00 MiB的显存,导致了显存不足的错误。
根据引用中的经验,为了减少中间数据的存储,您可以使用`torch.no_grad()`语句来避免存储过多的中间数据。此外,您还可以将模型和数据移动到GPU上进行计算,以减少数据传输和内存占用。以下是一个示例代码:
```
model = model.to(device)
with torch.no_grad():
img = Image.open(imgFile)
img = transform(img)
x = Variable(torch.unsqueeze(img, dim=0).float(), requires_grad=False)
x = x.to(device)
y = model(x).cpu()
```
引用中的文章也提到了类似的问题,并使用了上述的方法来解决显存不足的错误。通过使用`torch.no_grad()`语句,在进行特征提取时可以减少显存的使用。
希望这些信息能够帮助您解决CUDA显存不足的问题。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* *3* [GPU显存不足,报错:RuntimeError: CUDA out of memory.](https://blog.csdn.net/booklijian/article/details/113643387)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]
阅读全文
相关推荐











