nvidia errors in TensorFlow

最新推荐文章于 2024-05-29 08:09:47 发布

yusisc

最新推荐文章于 2024-05-29 08:09:47 发布

阅读量157

点赞数

分类专栏： python

本文链接：https://blog.csdn.net/yusisc/article/details/81364795

版权

python 专栏收录该内容

22 篇文章 0 订阅

订阅专栏

Go to check GPU memory usage, when encountering nvidia or cuda error.

I’ve been worked with tensorflow-gpu for a while. When run the codes, nvidia reports kinds of errors without clear instruction some time.

As far as my knowledge, met a nvidia error, one should check whether the gpu memory is occupied by other process by

nvidia-smi -l

As the following screenshot shows, the GPU-util is 0%, but the memory is nearly used out. If one run a new tensorflow process, errors are likely to be reported.
If you are coding with pycharm, there is scitific mode. In this mode, the process will not exit automatically, so the tensorflow session remains consuming the GPU memrory.

这里写图片描述