CUDA_ERROR_OUT_OF_MEMORY: out of memory

 

2023-02-04 22:17:02.457962: I tensorflow/stream_executor/cuda/cuda_driver.cc:831] failed to allocate 152.00M (159383552 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory

 

查看GPU内存使用情况：

nvidia-smi

 

 

 

命令行如何查看GPU的内存使用情况?

 

 

查看CPU内存使用情况

1、输入命令：top，

 

 

2、输入命令：htop，

 

 

若没有该命令，需要先安装，ubuntu安装命令为：
sudo apt-get install htop

centos安装命令为：
yum install htop

查看GPU内存使用情况
1、输入命令： nvidia-smi ，该命令只能显示一次。

2、实时显示
1）使用watch命令, 按照指定间隔刷新，例如下面命令是每隔1s刷新一次

watch -n 1 nvidia-smi

2）watch每个时间周期开启一个进程(PID)，查看后关闭进程，会影响cuda操作，建议使用nvidia-smi -l n或者nvidia-smi --loop=n，这个命令执行期间一直是一个进程PID。例如下面命令是每隔1s刷新一次。
nvidia-smi -l 1
————————————————
链接：https://blog.csdn.net/lansebingxuan/article/details/126105299