Tensorflow：UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed

Tensorflow报错解决：
sh
UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.

我在同一个进程中使用tensorflow和torch时，发生上述报错，同时显存也被占满。

有人说删除~/.nv/文件夹，这个文件夹是一些缓存信息，亲测无效.
还有人说要改tensorflow版本。。。懒得折腾了。。。
下面贴出我的解决方法，不用修改任何代码和tensorflow版本。
因为tensorflow在启动的时候会将所有显存占满，所以我们只用用import tensorflow前加上：

import os
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'

完美解决问题！