怎么释放gpu内存占用

释放gpu内存占用

有时在tensorflow调用完gpu后，内存并没有得到释放。这时候如果再次使用gpu训练tensorflow模型的话，就会导致一系列的问题。比如会报如下的错误：

could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

等等一些莫名的错误，这其实是gpu的内存被占用的原因，那这时候只需要把占用的内存释放就可以了。
两种方法可以解决：

1. 比较粗暴：直接重启电脑
2. 在终端运行

sudo rm -rf ~/.nv/ 

注：~/.nv/文件夹下是一些缓存信息，所以这一步的目的就是删除～/.nv/内的缓存信息。