Tried to allocate 154.00 MiB (GPU 0; 2.00 GiB total capacity; 1.08 GiB already allocated; 9.06 MiB free; 19.79 MiB cached)
时间: 2023-11-06 14:43:11 浏览: 61
这个错误提示意味着您尝试在GPU上分配154.00 MiB的内存,但GPU上总容量为2.00 GiB,已经分配了1.08 GiB的内存,只剩下很少的可用内存和一些被缓存的内存。您可以尝试减少您的模型大小或调整您的批处理大小,以减少内存使用量。您还可以考虑使用更大的GPU或多个GPU来处理更大的模型和数据。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![pptx](https://img-home.csdnimg.cn/images/20210720083543.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![m](https://img-home.csdnimg.cn/images/20210720083646.png)