pytorch 显存不够
时间: 2023-10-18 22:26:12 浏览: 38
如果PyTorch的显存不够,有几种方法可以尝试减少显存占用。其中一种有效的方法是将数据类型从float32转换为float16,这可以减少大约一半的显存占用。你可以在代码最前面添加"torch.set_default_dtype(torch.float16)"来实现。另外,如果在Python中调用PyTorch时,可能会出现显存和GPU占用不会自动释放的情况,此时你可以使用"torch.cuda.empty_cache()"来手动释放未使用的显存。这个函数会释放当前被缓存的未使用内存。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* [PyTorch 轻松节省显存的小技巧](https://blog.csdn.net/lucifer479/article/details/125849933)[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: 33.333333333333336%"]
- *2* [pytorch显存不足时的解决办法](https://blog.csdn.net/weixin_43914632/article/details/120759812)[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: 33.333333333333336%"]
- *3* [Pytorch释放显存占用方式](https://download.csdn.net/download/weixin_38652196/13760684)[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: 33.333333333333336%"]
[ .reference_list ]